Hi Tom, how can I implement bfloat16 in ATen with a packed_accessor? For floats the ATen definition and packed_accessor is (e.g.):
auto X = torch::zeros({a, b}, torch::dtype(torch::kF32))
X.packed_accessor32<float,2,torch::RestrictPtrTraits>()

but what should it read for bfloat16? I.e. what should come at the question marks below:
auto X = torch::zeros({a, b}, torch::dtype(torch::[?]))
X.packed_accessor32<[?],2,torch::RestrictPtrTraits>()

I can’t find the ATen reference for bfloat16 here, nor do I know what the packed_accessor reference should be…