Searched defs:BatchedTensorImpl (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchedTensorImpl.h | 42 struct TORCH_API BatchedTensorImpl : public c10::TensorImpl { struct
51 // BatchedTensorImpl wraps a Tensor argument
57 // bt = BatchedTensorImpl(ones(2, 3, 5, 7), lvl=1, dim=0) argument
102 // see NOTE: [BatchedTensorImpl levels invariant] argument
113 // BatchedTensorImpl. argument
|
| H A D | BatchedTensorImpl.cpp | 15 BatchedTensorImpl::BatchedTensorImpl(DispatchKeySet key_set, Tensor value, int64_t bdim, int64_t le… in BatchedTensorImpl() function in at::functorch::BatchedTensorImpl
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | LegacyBatchedTensorImpl.h | 58 struct TORCH_API BatchedTensorImpl : public c10::TensorImpl { struct
67 // BatchedTensorImpl wraps a Tensor argument
74 // bt = BatchedTensorImpl(ones(2, 3, 5, 7), [(lvl=1, dim=0), (lvl=2, argument
100 // Note: [BatchedTensorImpl levels invariant] argument
108 // BatchedTensorImpl. argument
|
| H A D | LegacyBatchedTensorImpl.cpp | 9 BatchedTensorImpl::BatchedTensorImpl(Tensor value, BatchDims bdims) in BatchedTensorImpl() function in at::BatchedTensorImpl
|