| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 44 WindowDimension* window_dim = default_conv_window_.add_dimensions(); in GpuConvRewriterTest() local
45 window_dim->set_size(1); in GpuConvRewriterTest()
46 window_dim->set_stride(1); in GpuConvRewriterTest()
47 window_dim->set_padding_low(0); in GpuConvRewriterTest()
48 window_dim->set_padding_high(0); in GpuConvRewriterTest()
49 window_dim->set_window_dilation(1); in GpuConvRewriterTest()
50 window_dim->set_base_dilation(1); in GpuConvRewriterTest()
316 const WindowDimension& window_dim = custom_call->window().dimensions(i); in TEST_F() local
319 EXPECT_EQ(3, window_dim.padding_low()); in TEST_F()
320 EXPECT_EQ(3, window_dim.padding_high()); in TEST_F()
[all …]
|
| D | gpu_conv_rewriter.cc | 111 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardFilter() local
112 if (window_dim.stride() != 1) { in MatchBackwardFilter()
118 if (window_dim.base_dilation() != 1) { in MatchBackwardFilter()
124 if (window_dim.padding_low() < 0) { in MatchBackwardFilter()
128 if (window_dim.window_reversal()) { in MatchBackwardFilter()
327 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardInput() local
328 if (window_dim.stride() != 1) { in MatchBackwardInput()
334 if (window_dim.window_dilation() != 1) { in MatchBackwardInput()
340 if (window_dim.window_reversal()) { in MatchBackwardInput()
|
| D | gpu_conv_padding_legalization.cc | 196 void IncreasePaddingLowBy(int64_t delta, WindowDimension* window_dim) { in IncreasePaddingLowBy() argument
197 window_dim->set_padding_low(window_dim->padding_low() + delta); in IncreasePaddingLowBy()
200 void IncreasePaddingHighBy(int64_t delta, WindowDimension* window_dim) { in IncreasePaddingHighBy() argument
201 window_dim->set_padding_high(window_dim->padding_high() + delta); in IncreasePaddingHighBy()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 298 auto window_dim = window.add_dimensions(); in HandleBatchGroupCount() local
299 window_dim->set_base_dilation(batch_group_count); in HandleBatchGroupCount()
300 window_dim->set_size(batch_group_count); in HandleBatchGroupCount()
301 window_dim->set_stride(batch_group_count - 1); in HandleBatchGroupCount()
302 window_dim->set_padding_low(0); in HandleBatchGroupCount()
303 window_dim->set_padding_high(0); in HandleBatchGroupCount()
304 window_dim->set_window_reversal(false); in HandleBatchGroupCount()
305 window_dim->set_window_dilation(1); in HandleBatchGroupCount()
653 auto window_dim = window.add_dimensions(); in HandleConvolution() local
654 window_dim->set_base_dilation(group_count); in HandleConvolution()
[all …]
|
| D | hlo_cse.cc | 116 for (const auto& window_dim : window_dims) { in AbslHashValue() local
117 h = H::combine(std::move(h), window_dim.size(), window_dim.stride(), in AbslHashValue()
118 window_dim.padding_low(), window_dim.padding_high(), in AbslHashValue()
119 window_dim.window_dilation(), window_dim.base_dilation(), in AbslHashValue()
120 window_dim.window_reversal()); in AbslHashValue()
|
| D | dynamic_padder.cc | 919 WindowDimension* window_dim = input_window->mutable_dimensions(dim_index); in RewriteInputWithDynamicPadding() local
922 window_dim->size(), window_dim->window_dilation()); in RewriteInputWithDynamicPadding()
929 padding_dim->set_edge_padding_high(window_dim->padding_high() + in RewriteInputWithDynamicPadding()
930 window_dim->padding_low()); in RewriteInputWithDynamicPadding()
931 padding_dim->set_interior_padding(window_dim->base_dilation() - 1); in RewriteInputWithDynamicPadding()
942 window_dim->padding_low() + in RewriteInputWithDynamicPadding()
944 window_dim->base_dilation()) + in RewriteInputWithDynamicPadding()
945 window_dim->padding_high(); in RewriteInputWithDynamicPadding()
946 window_dim->clear_padding_high(); in RewriteInputWithDynamicPadding()
947 window_dim->clear_padding_low(); in RewriteInputWithDynamicPadding()
[all …]
|
| D | hlo_cost_analysis.cc | 642 const auto& window_dim = window.dimensions(spatial_dimension); in GetConvolutionFlops() local
648 input_limits[spatial_dimension] == window_dim.base_dilation() && in GetConvolutionFlops()
649 window_dim.window_dilation() == 1 && in GetConvolutionFlops()
651 window_dim.stride() && in GetConvolutionFlops()
652 window_dim.padding_low() == 0 && window_dim.padding_high() == 0) { in GetConvolutionFlops()
659 window_dim.window_dilation() == 1 && window_dim.base_dilation() == 1 && in GetConvolutionFlops()
660 window_dim.stride() == 1 && in GetConvolutionFlops()
661 window_dim.padding_high() == output_limits[spatial_dimension] - 1 && in GetConvolutionFlops()
662 window_dim.padding_low() == output_limits[spatial_dimension] - 1) { in GetConvolutionFlops()
677 output_idx * window_dim.stride() - window_dim.padding_low() + in GetConvolutionFlops()
[all …]
|
| D | dynamic_dimension_inference.cc | 761 WindowDimension window_dim = hlo->window().dimensions(spatial_dim_index); in HandleDynamicConvolutionForward() local
763 dynamic_size, window_dim.size(), window_dim.window_dilation(), in HandleDynamicConvolutionForward()
764 window_dim.stride(), hlo->padding_type()); in HandleDynamicConvolutionForward()
765 TF_RET_CHECK(window_dim.base_dilation() == 1); in HandleDynamicConvolutionForward()
779 const WindowDimension& window_dim = window.dimensions(dimension); in HandleDynamicWindowSamePadding() local
780 if (!window_util::IsTrivialWindowDimension(window_dim)) { in HandleDynamicWindowSamePadding()
782 dynamic_size, window_dim.size(), window_dim.window_dilation(), in HandleDynamicWindowSamePadding()
783 window_dim.stride(), PaddingType::PADDING_SAME); in HandleDynamicWindowSamePadding()
1237 const WindowDimension& window_dim = in HandleReduceWindow() local
1245 if (!window_util::IsTrivialWindowDimension(window_dim)) { in HandleReduceWindow()
[all …]
|
| D | hlo_evaluator_typed_visitor.h | 1005 const auto& window_dim = window.dimensions(ki);
1007 out_index[output_spatial_dim] * window_dim.stride() -
1008 window_dim.padding_low() +
1009 rhs_spatial_index[ki] * window_dim.window_dilation();
1012 if (window_dim.base_dilation() > 1 &&
1013 undilated_index % window_dim.base_dilation() != 0) {
1020 if (window_dim.base_dilation() > 1) {
1021 lhs_spatial_index = undilated_index / window_dim.base_dilation();
1035 (window_dim.window_reversal()
1036 ? ((window_dim.size() - 1) - rhs_spatial_index[ki])
|
| D | scatter_expander.cc | 109 for (auto window_dim : update_window_dims) { in PermuteScatterAndWindowDims() local
110 permutation.push_back(window_dim); in PermuteScatterAndWindowDims()
|
| D | space_to_batch_converter.cc | 3403 auto window_dim = new_window.add_dimensions(); in PropagateOnBackpropFilterConv() local
3404 window_dim->set_base_dilation(1); in PropagateOnBackpropFilterConv()
3405 window_dim->set_size(1); in PropagateOnBackpropFilterConv()
3412 window_dim->set_stride(stride); in PropagateOnBackpropFilterConv()
3413 window_dim->set_padding_low(0); in PropagateOnBackpropFilterConv()
3414 window_dim->set_padding_high(0); in PropagateOnBackpropFilterConv()
3415 window_dim->set_window_reversal(false); in PropagateOnBackpropFilterConv()
3416 window_dim->set_window_dilation(1); in PropagateOnBackpropFilterConv()
|
| D | algebraic_simplifier.cc | 5718 for (const WindowDimension& window_dim : window.dimensions()) { in HandleReduceWindow() local
5720 padding_dim.set_edge_padding_low(window_dim.padding_low()); in HandleReduceWindow()
5721 padding_dim.set_edge_padding_high(window_dim.padding_high()); in HandleReduceWindow()
5915 auto& window_dim = *new_window.mutable_dimensions(i); in HandleReduceWindow() local
5916 window_dim.set_padding_low(window_dim.padding_low() + in HandleReduceWindow()
5917 window_dim.base_dilation() * in HandleReduceWindow()
5919 window_dim.set_padding_high(window_dim.padding_high() + in HandleReduceWindow()
5920 window_dim.base_dilation() * in HandleReduceWindow()
5923 CHECK_EQ(window_dim.base_dilation(), 1); in HandleReduceWindow()
5924 window_dim.set_base_dilation(1 + pad_dim.interior_padding()); in HandleReduceWindow()
|
| D | shape_inference.cc | 3579 for (int64_t window_dim : dim_numbers.update_window_dims()) { in ValidateScatterDimensionNumbers() local
3580 if (window_dim < 0 || window_dim >= updates_rank) { in ValidateScatterDimensionNumbers()
3584 updates_rank, window_dim); in ValidateScatterDimensionNumbers()
|
| D | elemental_ir_emitter.cc | 3085 const WindowDimension& window_dim) { in EmitConvolution() argument
3087 NSWMul(output_index, b_->getInt64(window_dim.stride())); in EmitConvolution()
3089 NSWMul(kernel_index, b_->getInt64(window_dim.window_dilation())); in EmitConvolution()
3091 b_->getInt64(window_dim.padding_low())); in EmitConvolution()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 306 const WindowDimension& window_dim = window.dimensions(i); in CreateNaiveMlirConv() local
308 builder.getAffineDimExpr(i + 2) * window_dim.stride() + in CreateNaiveMlirConv()
310 window_dim.padding_low()); in CreateNaiveMlirConv()
|