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()
522 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
748 const Window& conv_window, HloInstruction* original_hlo, SpmdBuilder* b) { in PartitionConvolutionTiledOutput()
824 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolutionBaseCase()
972 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolution()
Ddot_handler.cc449 const HloInstruction* original_hlo = nullptr, in GetWindowedEinsumConfiguration()
764 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionBaseCase()
2014 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionDotGroupOnBatch()
2358 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionDotGroupOnNonContracting()
2587 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionDotGroupOnContracting()
2797 const DotConvDimsMapping& dims_mapping, HloInstruction* original_hlo) { in ConvertDimsMappingWithFeatureGroupCount()
2820 const DotConvDimsMapping& dims_mapping, HloInstruction* original_hlo) { in ConvertDimsMappingWithBatchGroupCount()
3350 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionDot()
3738 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionDot()