Home
last modified time | relevance | path

Searched defs:result_dim (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dindexed_array_analysis.cc339 int64_t result_dim; member
376 for (int64_t result_dim = result_shape.size() - 1; result_dim >= 0; in ComputeReshapePassthroughDimPairs() local
783 auto map_passthrough_operand_dim_to_result_dim = [&](int64_t result_dim) { in FoldReshapeOfGatherNoDegenerateDims()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DNamedTensorUtils.cpp182 const auto result_dim = static_cast<int64_t>(result.dim()); in propagate_names_except() local
388 auto result_dim = result.dim(); in propagate_names_for_expand() local
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorConvolution.h35 const Index result_dim = input_dim - kernel_dim + 1; in IndexMapper() local