Searched defs:self_info (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | TriangularOps.cu | 127 auto self_info = cuda::detail::getTensorInfo<const scalar_t, int32_t>(self); in triu_tril_cuda_template() local
136 auto self_info = cuda::detail::getTensorInfo<const scalar_t, int64_t>(self); in triu_tril_cuda_template() local
|
| D | SortingCommon.cuh | 119 auto self_info = cuda::detail::getTensorInfo<const scalar_t, index_t>(self); in run_launcher() local
|
| D | Dropout.cu | 238 auto self_info = in launcher() local
|
| /external/pytorch/functorch/csrc/dim/ |
| D | dim.cpp | 2140 auto self_info = TensorInfo::create(A, self, false); in index() local
2339 TensorInfo self_info = TensorInfo::create(A, self, false, true); in getsetitem() local
2379 IndexingInfo getsetitem_flat(Arena& A, TensorInfo self_info, Slice<mpy::handle> input, Slice<DimEnt… in getsetitem_flat()
2790 auto self_info = TensorInfo::create(A, self, false); in py_split() local
|