Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc47 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolutionWithBatchGroupCount()
139 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolutionWithFeatureGroupCount()
233 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
520 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
746 const Window& conv_window, HloInstruction* original_hlo, SpmdBuilder* b) { in PartitionConvolutionTiledOutput()
822 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolutionBaseCase()
970 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolution()
Ddot_handler.cc473 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionBaseCase()
1657 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionDotGroupOnBatch()
1991 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionDotGroupOnNonContracting()
2086 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionDotGroupOnContracting()
2248 const DotConvDimsMapping& dims_mapping, HloInstruction* original_hlo) { in ConvertDimsMappingWithFeatureGroupCount()
2271 const DotConvDimsMapping& dims_mapping, HloInstruction* original_hlo) { in ConvertDimsMappingWithBatchGroupCount()
2438 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionDot()
2805 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionDot()