| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 43 WindowDimension* window_dim = default_conv_window_.add_dimensions(); in GpuConvRewriterTest() local
44 window_dim->set_size(1); in GpuConvRewriterTest()
45 window_dim->set_stride(1); in GpuConvRewriterTest()
46 window_dim->set_padding_low(0); in GpuConvRewriterTest()
47 window_dim->set_padding_high(0); in GpuConvRewriterTest()
48 window_dim->set_window_dilation(1); in GpuConvRewriterTest()
49 window_dim->set_base_dilation(1); in GpuConvRewriterTest()
315 const WindowDimension& window_dim = custom_call->window().dimensions(i); in TEST_F() local
318 EXPECT_EQ(3, window_dim.padding_low()); in TEST_F()
319 EXPECT_EQ(3, window_dim.padding_high()); in TEST_F()
[all …]
|
| D | gpu_conv_rewriter.cc | 190 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardFilter() local
191 if (window_dim.stride() != 1) { in MatchBackwardFilter()
197 if (window_dim.base_dilation() != 1) { in MatchBackwardFilter()
203 if (window_dim.padding_low() < 0) { in MatchBackwardFilter()
207 if (window_dim.window_reversal()) { in MatchBackwardFilter()
379 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardInput() local
380 if (window_dim.stride() != 1) { in MatchBackwardInput()
386 if (window_dim.window_dilation() != 1) { in MatchBackwardInput()
392 if (window_dim.window_reversal()) { in MatchBackwardInput()
|
| D | gpu_conv_padding_legalization.cc | 193 void IncreasePaddingLowBy(int64 delta, WindowDimension* window_dim) { in IncreasePaddingLowBy() argument
194 window_dim->set_padding_low(window_dim->padding_low() + delta); in IncreasePaddingLowBy()
197 void IncreasePaddingHighBy(int64 delta, WindowDimension* window_dim) { in IncreasePaddingHighBy() argument
198 window_dim->set_padding_high(window_dim->padding_high() + delta); in IncreasePaddingHighBy()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 293 auto window_dim = window.add_dimensions(); in HandleBatchGroupCount() local
294 window_dim->set_base_dilation(batch_group_count); in HandleBatchGroupCount()
295 window_dim->set_size(batch_group_count); in HandleBatchGroupCount()
296 window_dim->set_stride(batch_group_count - 1); in HandleBatchGroupCount()
297 window_dim->set_padding_low(0); in HandleBatchGroupCount()
298 window_dim->set_padding_high(0); in HandleBatchGroupCount()
299 window_dim->set_window_reversal(false); in HandleBatchGroupCount()
300 window_dim->set_window_dilation(1); in HandleBatchGroupCount()
646 auto window_dim = window.add_dimensions(); in HandleConvolution() local
647 window_dim->set_base_dilation(group_count); in HandleConvolution()
[all …]
|
| D | dynamic_padder.cc | 766 WindowDimension* window_dim = input_window->mutable_dimensions(dim_index); in RewriteInputWithDynamicPadding() local
769 window_dim->size(), window_dim->window_dilation()); in RewriteInputWithDynamicPadding()
776 padding_dim->set_edge_padding_high(window_dim->padding_high() + in RewriteInputWithDynamicPadding()
777 window_dim->padding_low()); in RewriteInputWithDynamicPadding()
778 padding_dim->set_interior_padding(window_dim->base_dilation() - 1); in RewriteInputWithDynamicPadding()
788 window_dim->padding_low() + in RewriteInputWithDynamicPadding()
790 window_dim->base_dilation()) + in RewriteInputWithDynamicPadding()
791 window_dim->padding_high(); in RewriteInputWithDynamicPadding()
792 window_dim->clear_padding_high(); in RewriteInputWithDynamicPadding()
793 window_dim->clear_padding_low(); in RewriteInputWithDynamicPadding()
[all …]
|
| D | hlo_cost_analysis.cc | 624 const auto& window_dim = window.dimensions(spatial_dimension); in HandleConvolution() local
630 input_limits[spatial_dimension] == window_dim.base_dilation() && in HandleConvolution()
631 window_dim.window_dilation() == 1 && in HandleConvolution()
633 window_dim.stride() && in HandleConvolution()
634 window_dim.padding_low() == 0 && window_dim.padding_high() == 0) { in HandleConvolution()
641 window_dim.window_dilation() == 1 && window_dim.base_dilation() == 1 && in HandleConvolution()
642 window_dim.stride() == 1 && in HandleConvolution()
643 window_dim.padding_high() == output_limits[spatial_dimension] - 1 && in HandleConvolution()
644 window_dim.padding_low() == output_limits[spatial_dimension] - 1) { in HandleConvolution()
658 const int64 undilated_index = output_idx * window_dim.stride() - in HandleConvolution()
[all …]
|
| D | dynamic_dimension_inference.cc | 713 WindowDimension window_dim = hlo->window().dimensions(spatial_dim_index); in HandleDynamicConvolutionForward() local
715 dynamic_size, window_dim.size(), window_dim.window_dilation(), in HandleDynamicConvolutionForward()
716 window_dim.stride(), hlo->padding_type()); in HandleDynamicConvolutionForward()
717 TF_RET_CHECK(window_dim.base_dilation() == 1); in HandleDynamicConvolutionForward()
731 const WindowDimension& window_dim = window.dimensions(dimension); in HandleDynamicWindowSamePadding() local
732 if (!window_util::IsTrivialWindowDimension(window_dim)) { in HandleDynamicWindowSamePadding()
734 dynamic_size, window_dim.size(), window_dim.window_dilation(), in HandleDynamicWindowSamePadding()
735 window_dim.stride(), PaddingType::PADDING_SAME); in HandleDynamicWindowSamePadding()
1064 const WindowDimension& window_dim = in HandleReduceWindow() local
1067 if (!window_util::IsTrivialWindowDimension(window_dim)) { in HandleReduceWindow()
[all …]
|
| D | scatter_expander.cc | 106 for (auto window_dim : update_window_dims) { in PermuteScatterAndWindowDims() local
107 permutation.push_back(window_dim); in PermuteScatterAndWindowDims()
|
| D | space_to_batch_converter.cc | 2520 auto window_dim = new_window.add_dimensions(); in PropagateOnBackpropFilterConv() local
2521 window_dim->set_base_dilation(1); in PropagateOnBackpropFilterConv()
2522 window_dim->set_size(1); in PropagateOnBackpropFilterConv()
2523 window_dim->set_stride(1); in PropagateOnBackpropFilterConv()
2524 window_dim->set_padding_low(0); in PropagateOnBackpropFilterConv()
2525 window_dim->set_padding_high(0); in PropagateOnBackpropFilterConv()
2526 window_dim->set_window_reversal(false); in PropagateOnBackpropFilterConv()
2527 window_dim->set_window_dilation(1); in PropagateOnBackpropFilterConv()
|
| D | hlo_evaluator_typed_visitor.h | 1185 const auto& window_dim = window.dimensions(ki);
1187 out_index[output_spatial_dim] * window_dim.stride() -
1188 window_dim.padding_low() +
1189 rhs_spatial_index[ki] * window_dim.window_dilation();
1192 if (window_dim.base_dilation() > 1 &&
1193 undilated_index % window_dim.base_dilation() != 0) {
1200 if (window_dim.base_dilation() > 1) {
1201 lhs_spatial_index = undilated_index / window_dim.base_dilation();
1215 (window_dim.window_reversal()
1216 ? ((window_dim.size() - 1) - rhs_spatial_index[ki])
|
| D | shape_inference.cc | 3460 for (int64 window_dim : dim_numbers.update_window_dims()) { in ValidateScatterDimensionNumbers() local
3461 if (window_dim < 0 || window_dim >= updates_rank) { in ValidateScatterDimensionNumbers()
3465 updates_rank, window_dim); in ValidateScatterDimensionNumbers()
|
| D | algebraic_simplifier.cc | 4975 auto& window_dim = *new_window.mutable_dimensions(i); in HandleReduceWindow() local
4976 window_dim.set_padding_low(window_dim.padding_low() + in HandleReduceWindow()
4978 window_dim.set_padding_high(window_dim.padding_high() + in HandleReduceWindow()
4981 CHECK_EQ(window_dim.base_dilation(), 1); in HandleReduceWindow()
4982 window_dim.set_base_dilation(1 + pad_dim.interior_padding()); in HandleReduceWindow()
|
| D | elemental_ir_emitter.cc | 2874 const WindowDimension& window_dim) { in EmitConvolution() argument
2876 NSWMul(output_index, b_->getInt64(window_dim.stride())); in EmitConvolution()
2878 NSWMul(kernel_index, b_->getInt64(window_dim.window_dilation())); in EmitConvolution()
2880 b_->getInt64(window_dim.padding_low())); in EmitConvolution()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 400 for (const auto& window_dim : reduce_window_op.window_dimensions()) { in CreateParallelLoopsToTraverseOutputAndWindow() local
404 rewriter->create<ConstantIndexOp>(loc, window_dim.getSExtValue())); in CreateParallelLoopsToTraverseOutputAndWindow()
556 for (const auto& window_dim : in InsertWindowLoops() local
558 Value upper = b->create<ConstantIndexOp>(loc, window_dim.getSExtValue()); in InsertWindowLoops()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 304 const WindowDimension& window_dim = window.dimensions(i); in CreateNaiveMlirConv() local
306 builder.getAffineDimExpr(i + 2) * window_dim.stride() + in CreateNaiveMlirConv()
308 window_dim.padding_low()); in CreateNaiveMlirConv()
|