Searched defs:physical_dim (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesScatterOps.cpp | 695 auto physical_dim = getPhysicalDim(self_, /*has_batch_dim*/true, dim); in scatter_batch_rule() local
733 auto physical_dim = getPhysicalDim(self_, /*has_batch_dim*/true, dim); in scatter_batch_rule() local
812 auto physical_dim = getPhysicalDim(self_, /*has_batch_dim*/true, dim); in gather_batch_rule() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | LegacyBatchingRegistrations.cpp | 375 auto physical_dim = getGradInputPhysicalDim(dim, input_sizes, grad_physical.numBatchDims()); in select_backward_batching_rule() local
395 auto physical_dim = getGradInputPhysicalDim(dim, input_sizes, grad_physical.numBatchDims()); in slice_backward_batching_rule() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | ir_array.cc | 390 int64_t physical_dim = logical_to_physical[dimension_mapping[i]]; in SourceIndexOfBroadcast() local
|