Searched refs:batchtensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/functorch/csrc/dim/ |
| H A D | dim.cpp | 179 const at::Tensor& batchtensor() { in batchtensor() function
305 return THPVariable_Wrap(self->batchtensor()); in Dim_get_batchtensor()
718 at::Tensor& batchtensor(Arena& A) { in batchtensor() function
799 …rInfo {t->tensor(A), t->levels(), t->has_device(), ensure_batched ? t->batchtensor(A) : TensorRef(… in create()
802 …->range(), Slice<DimEntry>(A, DimEntry(d)), false, ensure_batched ? d->batchtensor() : TensorRef()… in create()
1179 void inplace_update_layers(TensorRef batchtensor, Slice<DimEntry> levels) { in inplace_update_layers()
1181 auto impl = maybeGetBatchedImpl(*batchtensor); in inplace_update_layers()
1400 return THPVariable_Wrap(((Tensor*)self)->batchtensor(A)); }, NULL}, in __anon5de4c5480d02()
|