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.cc471 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionBaseCase() argument
1655 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionDotGroupOnBatch() argument
1989 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionDotGroupOnNonContracting() argument
2084 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionDotGroupOnContracting() argument
2436 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionDot() argument
2803 HloInstruction*, HloInstruction*, SpmdBuilder*, in PartitionDot() argument
2838 HloInstruction*, HloInstruction*, SpmdBuilder*, in HandleDotHelper() argument
Dspmd_partitioner.h72 SpmdBuilder(const std::string& name, HloInstruction* hlo) in SpmdBuilder() function