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.cc512 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionBaseCase()
1738 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionDotGroupOnBatch()
2001 bool lhs_matching, const Shape& output_base_shape, const Shape& other_shape, in GetNonContractingPartitionGroupedShardingForOtherOperand()
2077 const Shape& output_base_shape, const HloSharding& output_sharding, in PartitionDotGroupOnNonContracting()
2169 const Shape& output_base_shape, const HloSharding& output_sharding, in GetDotGroupPartitionContractingOutputShardings()
2265 const Shape& output_base_shape, const HloSharding& output_sharding, in PartitionDotGroupOnContracting()
2483 const PartitionedHlo& rhs, const Shape& output_base_shape, in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
2617 const PartitionedHlo& rhs, const Shape& output_base_shape, in PrioritizeContractingDimensionsPartitioning()
2764 const PartitionedHlo& rhs, const Shape& output_base_shape, in LhsIsBestMatchForNonContractingPartitioning()
2820 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionDot()
[all …]