Home
last modified time | relevance | path

Searched defs:conv_window (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_rewriter_test.cc107 Window conv_window = default_conv_window_; in TEST_F() local
149 Window conv_window = default_conv_window_; in TEST_F() local
181 Window conv_window = default_conv_window_; in TEST_F() local
211 Window conv_window = default_conv_window_; in TEST_F() local
240 Window conv_window = default_conv_window_; in TEST_F() local
271 Window conv_window = default_conv_window_; in TEST_F() local
340 Window conv_window = default_conv_window_; in TEST_F() local
424 Window conv_window = default_conv_window_; in TEST_F() local
476 Window conv_window = default_conv_window_; in TEST_F() local
532 Window conv_window = default_conv_window_; in TEST_F() local
[all …]
H A Dgpu_conv_padding_legalization.cc50 Window* conv_window, const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedAndSlicedInput()
125 HloInstruction* MaybePaddedKernel(const Window& conv_window, in MaybePaddedKernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc46 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithBatchGroupCount()
138 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithFeatureGroupCount()
232 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
521 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
747 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionTiledOutput()
823 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionBaseCase()
899 const Window& conv_window) { in CreateShardedConvConvolution()
971 const Window& conv_window)>& create_sharded_conv, in PartitionConvolution()
1033 const Window& conv_window) -> StatusOr<HloInstruction*> { in HandleConvolution()
H A Ddot_handler.cc89 const Window& conv_window) -> StatusOr<HloInstruction*> { in HandleDot()
447 const Window& conv_window, const DotConvDimsMapping& dims_mapping, in GetWindowedEinsumConfiguration()
763 const Window& conv_window)>& create_sharded_dot, in PartitionBaseCase()
2013 const Window& conv_window)>& create_sharded_dot, in PartitionDotGroupOnBatch()
2357 const Window& conv_window)>& create_sharded_dot, in PartitionDotGroupOnNonContracting()
2586 const Window& conv_window)>& create_sharded_dot, in PartitionDotGroupOnContracting()
2687 const Window& conv_window) -> StatusOr<HloInstruction*> { in PartitionDotGroupOnContracting()
2851 int64_t rhs_batch_partitions, const Window& conv_window) { in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
2997 const Window& conv_window, in PrioritizeContractingDimensionsPartitioning()
3224 int64_t rhs_batch_partitions, SpmdBuilder* b, const Window& conv_window, in LhsIsBestMatchForNonContractingPartitioning()
[all …]