Searched defs:trailing_dims (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_glu.cpp | 55 size_t trailing_dims = getTrailingDims(in, dim); in mul_tensors() local
92 size_t trailing_dims = getTrailingDims(in, dim); in slice_tensor() local
|
| H A D | op_unbind_copy.cpp | 51 const size_t trailing_dims = getTrailingDims(input, dim); in unbind_copy_int_out() local
|
| H A D | op_index_select.cpp | 57 size_t trailing_dims = getTrailingDims(in, dim); in index_select_out() local
|
| H A D | op_split_copy.cpp | 55 const size_t trailing_dims = getTrailingDims(input, dim); in split_copy_Tensor_out() local
|
| H A D | op_slice_scatter.cpp | 72 size_t trailing_dims = getTrailingDims(input, dim); in slice_scatter_out() local
|
| H A D | op_cumsum.cpp | 59 const size_t trailing_dims = getTrailingDims(self, dim); in cumsum_tensors() local
|
| H A D | op_split_with_sizes_copy.cpp | 68 const size_t trailing_dims = getTrailingDims(in, dim); in split_with_sizes_copy_out() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | copy_ops_util.h | 36 size_t trailing_dims = getTrailingDims(out, dim); in _as_strided_copy() local
|
| H A D | select_copy_util.cpp | 58 size_t trailing_dims = getTrailingDims(in, dim); in select_copy_util() local
|
| H A D | slice_util.cpp | 160 size_t trailing_dims = getTrailingDims(in, dim); in compute_slice() local
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | nn_utils.cpp | 611 std::vector<int64_t> trailing_dims(num_dim, 4); in TEST_F() local
850 std::vector<int64_t> trailing_dims(num_dim, 4); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PackedSequence.cpp | 211 IntArrayRef trailing_dims = max_size.slice(1); in pad_sequence() local
|