Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter_test.cc106 Window conv_window = default_conv_window_; in TEST_F() local
148 Window conv_window = default_conv_window_; in TEST_F() local
180 Window conv_window = default_conv_window_; in TEST_F() local
210 Window conv_window = default_conv_window_; in TEST_F() local
239 Window conv_window = default_conv_window_; in TEST_F() local
270 Window conv_window = default_conv_window_; in TEST_F() local
339 Window conv_window = default_conv_window_; in TEST_F() local
423 Window conv_window = default_conv_window_; in TEST_F() local
475 Window conv_window = default_conv_window_; in TEST_F() local
531 Window conv_window = default_conv_window_; in TEST_F() local
[all …]
Dgpu_conv_padding_legalization.cc49 Window* conv_window, const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedAndSlicedInput()
123 HloInstruction* MaybePaddedKernel(const Window& conv_window, in MaybePaddedKernel()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
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()
519 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
745 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionTiledOutput()
821 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionBaseCase()
897 const Window& conv_window) { in CreateShardedConvConvolution()
969 const Window& conv_window)>& create_sharded_conv, in PartitionConvolution()
1028 const Window& conv_window) -> StatusOr<HloInstruction*> { in HandleConvolution()
Ddot_handler.cc80 const Window& conv_window) -> StatusOr<HloInstruction*> { in HandleDot()
472 const Window& conv_window)>& create_sharded_dot, in PartitionBaseCase()
1656 const Window& conv_window)>& create_sharded_dot, in PartitionDotGroupOnBatch()
1990 const Window& conv_window)>& create_sharded_dot, in PartitionDotGroupOnNonContracting()
2085 const Window& conv_window)>& create_sharded_dot, in PartitionDotGroupOnContracting()
2437 const Window& conv_window)>& create_sharded_dot, in PartitionDot()
2804 const Window& conv_window)>& create_sharded_dot, in PartitionDot()
2842 Window conv_window; in HandleDotHelper() local