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()
1031 const Window& conv_window) -> StatusOr<HloInstruction*> { in HandleConvolution()
Ddot_handler.cc80 const Window& conv_window) -> StatusOr<HloInstruction*> { in HandleDot()
517 const Window& conv_window)>& create_sharded_dot, in PartitionBaseCase()
1745 const Window& conv_window)>& create_sharded_dot, in PartitionDotGroupOnBatch()
2081 const Window& conv_window)>& create_sharded_dot, in PartitionDotGroupOnNonContracting()
2269 const Window& conv_window)>& create_sharded_dot, in PartitionDotGroupOnContracting()
2368 const Window& conv_window) -> StatusOr<HloInstruction*> { in PartitionDotGroupOnContracting()
2825 const Window& conv_window)>& create_sharded_dot, in PartitionDot()
3176 const Window& conv_window)>& create_sharded_dot, in PartitionDot()
3217 Window conv_window; in HandleDotHelper() local