| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | transpose.cc | 554 int b_dim = inv_permutation[a_dim]; in RemoveTrivialDimensions() local
572 for (int b_dim = 0; b_dim < ndim; ++b_dim) { in RemoveTrivialDimensions() local
635 for (int b_dim = 0; b_dim < ndim; ++b_dim) { in CoalesceDimensions() local
768 int b_dim = inverse_permutation[a_dim]; in BuildPlanNodes() local
1156 int b_dim = inverse_permutation[l.dim_in_a]; in Initialize() local
1223 int b_dim = inverse_permutation[a_dim]; in ChooseParallelizationStrategy() local
|
| /aosp_15_r20/external/libdav1d/src/ |
| H A D | decode.c | 450 const uint8_t *const b_dim = dav1d_block_dimensions[bs]; in read_vartx_tree() local
594 const uint8_t *const b_dim, in affine_lowest_px()
617 const uint8_t *const b_dim, in affine_lowest_px_luma()
624 const uint8_t *const b_dim, in affine_lowest_px_chroma()
637 const uint8_t *const b_dim, in obmc_lowest_px()
692 const uint8_t *const b_dim = dav1d_block_dimensions[bs]; in decode_b() local
2082 const uint8_t *const b_dim = dav1d_block_dimensions[bs]; in checked_decode_b() local
|
| H A D | recon_tmpl.c | 66 const uint8_t *const b_dim = dav1d_block_dimensions[bs]; in get_skip_ctx() local
835 const uint8_t *const b_dim = dav1d_block_dimensions[bs]; in bytefn() local
1057 const uint8_t *const b_dim, const int pl, in obmc()
1120 const uint8_t *const b_dim, const int pl, in warp_affine()
1190 const uint8_t *const b_dim = dav1d_block_dimensions[bs]; in bytefn() local
1570 const uint8_t *const b_dim = dav1d_block_dimensions[bs]; in bytefn() local
|
| H A D | lf_mask.c | 272 const uint8_t *const b_dim = dav1d_block_dimensions[bs]; in dav1d_create_lf_mask_intra() local
335 const uint8_t *const b_dim = dav1d_block_dimensions[bs]; in dav1d_create_lf_mask_inter() local
|
| H A D | refmvs.c | 356 const uint8_t *const b_dim = dav1d_block_dimensions[bs]; in dav1d_refmvs_find() 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 | 114 auto b_dim = b_shape.size(); in tensors_are_broadcastable_between() local
222 auto b_dim = b_size.size(); in get_broadcast_target_size() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator.cc | 1309 int b_dim = bigger_rank_shape->dim(b_i).size(); in CountBatchMatMulOperations() local
|