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> lhs_to_rhs_indices(output_base_shape.rank()); in PartitionConvolutionWithBatchGroupCount() local
169 std::vector<int64> lhs_to_rhs_indices(output_base_shape.rank()); in PartitionConvolutionWithFeatureGroupCount() local
249 std::vector<int64> lhs_to_rhs_indices(output_base_shape.rank()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
539 std::vector<int64> lhs_to_rhs_indices(output_base_shape.rank()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
Ddot_handler.cc105 std::vector<int64> lhs_to_rhs_indices; member
342 std::vector<int64> lhs_to_rhs_indices(lhs_rank, -1); in ComputeDimensionIndexMapping() local