Home
last modified time | relevance | path

Searched refs:conv_window (Results 1 – 6 of 6) 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
107 conv_window.mutable_dimensions(1)->set_size(2); in TEST_F()
108 conv_window.mutable_dimensions(1)->set_window_dilation(2); in TEST_F()
112 /*batch_group_count=*/1, conv_window, in TEST_F()
117 /*batch_group_count=*/1, conv_window, in TEST_F()
148 Window conv_window = default_conv_window_; in TEST_F() local
149 conv_window.mutable_dimensions(1)->set_size(3); in TEST_F()
153 /*batch_group_count=*/1, conv_window, in TEST_F()
158 /*batch_group_count=*/1, conv_window, in TEST_F()
180 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() argument
52 if (!window_util::HasSymmetricPadding(*conv_window) || in MaybePaddedAndSlicedInput()
53 window_util::HasBaseDilation(*conv_window)) { in MaybePaddedAndSlicedInput()
66 if (conv_window->dimensions(i).padding_low() > 0) { in MaybePaddedAndSlicedInput()
68 conv_window->dimensions(i).padding_low()); in MaybePaddedAndSlicedInput()
69 conv_window->mutable_dimensions(i)->set_padding_low(0); in MaybePaddedAndSlicedInput()
71 if (conv_window->dimensions(i).padding_high() > 0) { in MaybePaddedAndSlicedInput()
73 conv_window->dimensions(i).padding_high()); in MaybePaddedAndSlicedInput()
74 conv_window->mutable_dimensions(i)->set_padding_high(0); in MaybePaddedAndSlicedInput()
76 if (conv_window->dimensions(i).base_dilation() != 1) { in MaybePaddedAndSlicedInput()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc46 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithBatchGroupCount() argument
47 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolutionWithBatchGroupCount()
125 create_sharded_conv(lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionConvolutionWithBatchGroupCount()
138 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithFeatureGroupCount() argument
139 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolutionWithFeatureGroupCount()
217 create_sharded_conv(lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionConvolutionWithFeatureGroupCount()
232 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() argument
233 const Window& conv_window, HloInstruction* original_hlo, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
326 const auto& wd = conv_window.dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
344 Window new_window = conv_window; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
[all …]
Dconvolution_handler.h34 const Window& conv_window)>& create_sharded_conv,
35 const Window& conv_window, HloInstruction* original_hlo,
Ddot_handler.cc80 const Window& conv_window) -> StatusOr<HloInstruction*> { in HandleDot() argument
472 const Window& conv_window)>& create_sharded_dot, in PartitionBaseCase() argument
473 const Window& conv_window, HloModule* module, HloInstruction* original_hlo, in PartitionBaseCase()
525 auto dot, create_sharded_dot(lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionBaseCase()
552 create_sharded_dot(lhs.hlo(), resharded_rhs.hlo(), b, conv_window)); in PartitionBaseCase()
569 create_sharded_dot(resharded_lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionBaseCase()
933 &body_b, conv_window)); in PartitionBaseCase()
1105 auto dot, create_sharded_dot(dot_lhs, dot_rhs, &body_b, conv_window)); in PartitionBaseCase()
1512 auto dot, create_sharded_dot(lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionBaseCase()
1534 b, conv_window)); in PartitionBaseCase()
[all …]
Dspmd_partitioner.h446 const Window& conv_window)>& create_sharded_dot);