Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc42 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolutionWithBatchGroupCount()
134 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolutionWithFeatureGroupCount()
228 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
515 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
741 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolutionTiledOutput()
817 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolutionBaseCase()
965 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolution()
Ddot_handler.cc467 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionBaseCase()
1649 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionDotGroupOnBatch()
1912 bool lhs_matching, const Shape& output_base_shape, const Shape& other_shape, in GetNonContractingPartitionGroupedShardingForOtherOperand()
1986 const Shape& output_base_shape, const HloSharding& output_sharding, in PartitionDotGroupOnNonContracting()
2080 int64 output_rhs_non_contracting_partitions, const Shape& output_base_shape, in PartitionDotGroupOnContracting()
2289 const PartitionedHlo& rhs, const Shape& output_base_shape, in LhsIsBestMatchForNonContractingPartitioning()
2432 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionDot()
2799 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionDot()