Searched defs:self_shape (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorMatmul.cu | 310 …const IntArrayRef &self_shape = get_size_for_index(self, i), &mat2_shape = get_size_for_index(mat2… in bmm_nested_cuda() local
354 const IntArrayRef& self_shape = get_size_for_index(self, i); in bmm_nested_cuda() local
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | flop_counter.py | 67 def addmm_flop(self_shape, a_shape, b_shape, out_shape=None, **kwargs) -> int: argument
85 def baddbmm_flop(self_shape, a_shape, b_shape, out_shape=None, **kwargs) -> int: argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qmul.cpp | 139 const auto self_shape = xnnp_utils::get_mem_format_aware_shape(self_contig); in _mul_out_xnnpack() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorMatmul.cpp | 43 const IntArrayRef& self_shape = get_size_for_index(self, i); in bmm_nested() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner_util.cc | 790 auto self_shape = hlo->shape(); in ExchangeHalo() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 86 def check_inplace_broadcast(self_shape, *args_shape): argument
|