Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc67 std::vector<int64> rhs_to_lhs_indices(output_base_shape.rank()); in PartitionConvolutionWithBatchGroupCount() local
158 std::vector<int64> rhs_to_lhs_indices(output_base_shape.rank()); in PartitionConvolutionWithFeatureGroupCount() local
240 std::vector<int64> rhs_to_lhs_indices(output_base_shape.rank()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
530 std::vector<int64> rhs_to_lhs_indices(output_base_shape.rank()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
Ddot_handler.cc107 std::vector<int64> rhs_to_lhs_indices; member
344 std::vector<int64> rhs_to_lhs_indices(rhs_rank, -1); in ComputeDimensionIndexMapping() local