Home
last modified time | relevance | path

Searched defs:SpmdBuilder (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc45 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionConvolutionWithBatchGroupCount() argument
137 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionConvolutionWithFeatureGroupCount() argument
231 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() argument
518 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() argument
744 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionConvolutionTiledOutput() argument
820 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionConvolutionBaseCase() argument
968 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionConvolution() argument
Ddot_handler.cc516 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionBaseCase() argument
1744 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionDotGroupOnBatch() argument
2080 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionDotGroupOnNonContracting() argument
2268 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionDotGroupOnContracting() argument
2824 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionDot() argument
3175 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionDot() argument
3210 HloInstruction*, HloInstruction*, SpmdBuilder*, in HandleDotHelper() argument
Dspmd_partitioner.h78 SpmdBuilder(const std::string& name, HloInstruction* hlo) in SpmdBuilder() function