Searched defs:a_dim (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | transpose.cc | 553 for (int a_dim = 0; a_dim < ndim; ++a_dim) { in RemoveTrivialDimensions() local
573 int a_dim = permutation[b_dim]; in RemoveTrivialDimensions() local
609 for (int a_dim = 0; a_dim < ndim; ++a_dim) { in CoalesceDimensions() local
636 int a_dim = permutation[b_dim]; in CoalesceDimensions() local
767 int a_dim = loop.dim_in_a; in BuildPlanNodes() local
1222 int a_dim = loop.dim_in_a; in ChooseParallelizationStrategy() local
|
| /aosp_15_r20/external/executorch/backends/cadence/hifi/operators/ |
| H A D | op_div.cpp | 75 int a_dim = a.dim(), b_dim = b.dim(), out_dim = out.dim(); in div_out() local
189 int a_dim = a.dim(), b_dim = b.dim(), out_dim = out.dim(); in div_out_mode() local
|
| H A D | op_mul.cpp | 96 int a_dim = a.dim(), b_dim = b.dim(), out_dim = out.dim(); in mul_out() local
|
| H A D | op_where.cpp | 54 int a_dim = a.dim(), b_dim = b.dim(), con_dim = cond.dim(), in where_out() local
|
| H A D | op_add.cpp | 127 int a_dim = a.dim(), b_dim = b.dim(), out_dim = out.dim(); in add_out() local
|
| H A D | op_sub.cpp | 122 int a_dim = a.dim(), b_dim = b.dim(), out_dim = out.dim(); in sub_out() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | broadcast_util.cpp | 113 auto a_dim = a_shape.size(); in tensors_are_broadcastable_between() local
221 auto a_dim = a_size.size(); in get_broadcast_target_size() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | CrossKernel.cpp | 29 const int64_t a_dim = a.dim(); in apply_cross() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Loss.cpp | 201 auto a_dim = anchor.dim(); in triplet_margin_loss() local
|
| /aosp_15_r20/external/executorch/runtime/core/exec_aten/util/ |
| H A D | tensor_util.h | 789 size_t a_dim = static_cast<size_t>(a.dim()); in tensor_has_expected_size() local
|