output_base_shape (definition) in projects: aosp_15_r20

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A D	convolution_handler.cc	42 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() 517 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() 743 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolutionTiledOutput() 819 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolutionBaseCase() 967 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionConvolution()
H A D	dot_handler.cc	758 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionBaseCase() 2006 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionDotGroupOnBatch() 2275 bool lhs_matching, const Shape& output_base_shape, const Shape& other_shape, in GetNonContractingPartitionGroupedShardingForOtherOperand() 2353 const Shape& output_base_shape, const HloSharding& output_sharding, in PartitionDotGroupOnNonContracting() 2460 const Shape& output_base_shape, const HloSharding& output_sharding, in GetDotGroupPartitionContractingOutputShardings() 2582 const Shape& output_base_shape, const HloSharding& output_sharding, in PartitionDotGroupOnContracting() 2843 const PartitionedHlo& rhs, const Shape& output_base_shape, in EstimateWindowedEinsumIterationsForNonContractingPartitioning() 2988 const PartitionedHlo& rhs, const Shape& output_base_shape, in PrioritizeContractingDimensionsPartitioning() 3216 const PartitionedHlo& rhs, const Shape& output_base_shape, in LhsIsBestMatchForNonContractingPartitioning() 3344 PartitionedHlo lhs, PartitionedHlo rhs, const Shape& output_base_shape, in PartitionDot() [all …]