Home
last modified time | relevance | path

Searched defs:lhs_to_rhs_indices (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc78 std::vector<int64_t> lhs_to_rhs_indices(output_base_shape.rank()); in PartitionConvolutionWithBatchGroupCount() local
169 std::vector<int64_t> lhs_to_rhs_indices(output_base_shape.rank()); in PartitionConvolutionWithFeatureGroupCount() local
249 std::vector<int64_t> lhs_to_rhs_indices(output_base_shape.rank()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
541 std::vector<int64_t> lhs_to_rhs_indices(output_base_shape.rank()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
Ddot_handler.cc114 std::vector<int64_t> lhs_to_rhs_indices; member
345 std::vector<int64_t> lhs_to_rhs_indices(lhs_rank, -1); in ComputeDimensionIndexMapping() local