Home
last modified time | relevance | path

Searched refs:padding_low (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc66 if (conv_window->dimensions(i).padding_low() > 0) { in MaybePaddedAndSlicedInput()
68 conv_window->dimensions(i).padding_low()); in MaybePaddedAndSlicedInput()
102 if (conv_window->dimensions(i).padding_low() < 0) { in MaybePaddedAndSlicedInput()
103 start_indices[dim] += -conv_window->dimensions(i).padding_low(); in MaybePaddedAndSlicedInput()
194 window_dim->set_padding_low(window_dim->padding_low() + delta); in IncreasePaddingLowBy()
226 int64 padding_low = backward_conv->window().dimensions(i).padding_low(); in CanonicalizeBackwardFilterConvolution() local
228 if (padding_low < 0 || padding_high < 0) { in CanonicalizeBackwardFilterConvolution()
235 int64 new_conv_padding = std::min(padding_low, padding_high); in CanonicalizeBackwardFilterConvolution()
238 padding_low - new_conv_padding); in CanonicalizeBackwardFilterConvolution()
291 int64 padding_low = backward_conv->window().dimensions(i).padding_low(); in CanonicalizeBackwardInputConvolution() local
[all …]
Dgpu_conv_rewriter.cc203 if (window_dim.padding_low() < 0) { in MatchBackwardFilter()
251 dim->set_padding_low(conv->window().dimensions(i).padding_low()); in MatchBackwardFilter()
270 padded_input_size - input_size - dim->padding_low(); in MatchBackwardFilter()
272 CHECK_GE(dim->padding_low(), 0); in MatchBackwardFilter()
280 if (dim->padding_low() >= min_padding_high && in MatchBackwardFilter()
281 dim->padding_low() <= max_padding_high) { in MatchBackwardFilter()
282 dim->set_padding_high(dim->padding_low()); in MatchBackwardFilter()
284 if (dim->padding_low() < min_padding_high) { in MatchBackwardFilter()
420 kernel_size - 1 - old_window.dimensions(i).padding_low(); in MatchBackwardInput()
Dgpu_conv_rewriter_test.cc318 EXPECT_EQ(3, window_dim.padding_low()); in TEST_F()
455 EXPECT_EQ(0, window_dim.padding_low()); in TEST_F()
559 EXPECT_EQ(0, backward_conv_col_dim.padding_low()); in TEST_F()
Dgpu_conv_runner.cc352 CHECK_EQ(dim.padding_low(), dim.padding_high()) << inst_as_string; in GetGpuConvConfig()
407 window.dimensions(dim).padding_low()) in GetGpuConvConfig()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_pad_op.cc37 std::vector<int64> padding_low; in Compile() local
41 &padding_low)); in Compile()
50 OP_REQUIRES(context, rank == padding_low.size(), in Compile()
54 padding_low.size(), " vs. ", rank, ")")); in Compile()
75 dim->set_edge_padding_low(padding_low[i]); in Compile()
Dmatrix_diag_ops.cc204 int padding_low = length_to_pad_to - max_diag_len; in SetMatrixDiag() local
208 std::swap(padding_low, padding_high); in SetMatrixDiag()
211 ->set_edge_padding_low(padding_low); in SetMatrixDiag()
/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.cc72 if (dim.padding_low() != 0) { in ToString()
73 StrAppend(&str, ",padding_low=", dim.padding_low()); in ToString()
117 return StrCat(dim.padding_low(), "_", dim.padding_high()); in ToString()
149 if (dim.padding_low() != 0 || dim.padding_high() != 0) { in HasPadding()
158 return dim.padding_low() == dim.padding_high(); in HasSymmetricPadding()
172 return dim.padding_low() < 0 || dim.padding_high() < 0; in HasNegativePadding()
219 window_dimension.padding_low() == 0 && in IsTrivialWindowDimension()
Dxla_data.proto436 int64 padding_low = 3; field
438 // As padding_low, but on the high end of this dimension. For example, in the
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_as_convolution_util.cc36 wd.padding_high() == 0 && wd.padding_low() == 0 && in ConvSpatialDimensionIsParallel()
43 wd.padding_low() == lhs_size - 1 && wd.window_reversal() && in ConvSpatialDimensionIsParallel()
78 wd.padding_low() == 0 && !wd.window_reversal()) { in ParseConvolutionDimsInfo()
86 wd.padding_low() == 0 && !wd.window_reversal()) { in ParseConvolutionDimsInfo()
92 wd.padding_low() == rhs_size - 1 && wd.window_reversal()) { in ParseConvolutionDimsInfo()
Dconvolution_4d_expander.cc48 instruction->window().dimensions(i).padding_low() == 0 && in InstructionMatchesPattern()
79 instruction->window().dimensions(i).padding_low() == 0 && in ExpandInstruction()
Dhlo_cost_analysis.cc634 window_dim.padding_low() == 0 && window_dim.padding_high() == 0) { in HandleConvolution()
644 window_dim.padding_low() == output_limits[spatial_dimension] - 1) { in HandleConvolution()
659 window_dim.padding_low() + in HandleConvolution()
Dspace_to_batch_converter.cc302 .padding_low(); in IsConvSuitableForSpaceToBatch()
982 consumer->window().dimensions(old_space_dim).padding_low(); in CanPropagate()
1167 if (window.dimensions(old_space_dim).padding_low() != 0) { in SupportedOpForPropagation()
1455 consumer->window().dimensions(dim).padding_low()); in Propagate()
2404 .padding_low(); in PropagateOnBackpropFilterConv()
2610 .padding_low(); in GetConvolutionDetails()
Dalgebraic_simplifier_test.cc4409 EXPECT_EQ(root->window().dimensions(0).padding_low(), 10); in TEST_F()
4410 EXPECT_EQ(root->window().dimensions(1).padding_low(), 11); in TEST_F()
4411 EXPECT_EQ(root->window().dimensions(2).padding_low(), 10); in TEST_F()
4412 EXPECT_EQ(root->window().dimensions(3).padding_low(), 10); in TEST_F()
4495 EXPECT_EQ(root->window().dimensions(0).padding_low(), 10); in TEST_F()
4496 EXPECT_EQ(root->window().dimensions(1).padding_low(), 11); in TEST_F()
4497 EXPECT_EQ(root->window().dimensions(2).padding_low(), 10); in TEST_F()
4498 EXPECT_EQ(root->window().dimensions(3).padding_low(), 10); in TEST_F()
7149 EXPECT_EQ(conv->window().dimensions(0).padding_low(), 1); in TEST_F()
7150 EXPECT_EQ(conv->window().dimensions(1).padding_low(), 1); in TEST_F()
Ddynamic_padder.cc777 window_dim->padding_low()); in RewriteInputWithDynamicPadding()
788 window_dim->padding_low() + in RewriteInputWithDynamicPadding()
Dalgebraic_simplifier.cc4976 window_dim.set_padding_low(window_dim.padding_low() + in HandleReduceWindow()
5183 (w.padding_low() != 0 || w.padding_high() != 0 || in FoldConvInputPad()
5193 w.set_padding_low(w.padding_low() + p.edge_padding_low()); in FoldConvInputPad()
5299 window_dims[spatial_dim].padding_low() == 0 && in SwapConvOperands()
5342 window_dims[spatial_dim].padding_low() - in SwapConvOperands()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py222 padding_low=[2, 1],
241 padding_low=[0, -1],
258 padding_low=[2, 1],
267 padding_low=[2, -2],
277 padding_low=[2, 1],
288 padding_low=[0, 0],
297 padding_low=[0, 0, 0],
306 padding_low=[0, 0],
315 padding_low=[0, 0],
325 padding_low=[0, 0],
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc373 int64 padding_low = wd.padding_low(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
376 int64 window_count = 1 + (padding_low + padding_high + base - in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
382 (window_count - 1) * wd.stride() - padding_low + in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
389 padding_low + wd.window_dilation() - 1, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
417 -padding_low, 1)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
670 int64 padding_low = wd.padding_low(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
673 int64 window_count = 1 + (padding_low + padding_high + base - in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
681 lhs_shard_size - rhs_shard_size * wd.window_dilation(), padding_low, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
688 padding_low, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
693 int64 explicit_left_padding_on_full_shape = padding_low; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
[all …]
Dspmd_partitioner.cc595 wd.padding_high() + wd.padding_low(); in ReshardAsWindowedInput()
615 explicit_left_padding[i] = wd.padding_low() / wd.base_dilation(); in ReshardAsWindowedInput()
616 swd->set_padding_low(wd.padding_low() % wd.base_dilation()); in ReshardAsWindowedInput()
624 wd.base_dilation() - 1 - swd->padding_low(), wd.base_dilation()); in ReshardAsWindowedInput()
629 dilated_shard_size + wd.base_dilation() - 1 - swd->padding_low(), in ReshardAsWindowedInput()
652 swd->padding_low() - in ReshardAsWindowedInput()
656 swd->padding_low()); in ReshardAsWindowedInput()
680 if (start != swd->padding_low()) { in ReshardAsWindowedInput()
695 wd.base_dilation(), swd->padding_low() - max_pad_low, 1) in ReshardAsWindowedInput()
1617 start_indices[i] = -dim.padding_low(); in HandleSlice()
[all …]
Dspmd_partitioner_test.cc1238 EXPECT_EQ(root->window().dimensions(0).padding_low(), 1); in TEST_F()
1298 EXPECT_EQ(root->operand(0)->window().dimensions(0).padding_low(), 1); in TEST_F()
1349 EXPECT_EQ(root->window().dimensions(0).padding_low(), 0); in TEST_F()
1403 EXPECT_EQ(root->window().dimensions(0).padding_low(), 0); in TEST_F()
1496 EXPECT_EQ(root->operand(0)->window().dimensions(0).padding_low(), 0); in TEST_F()
3421 EXPECT_EQ(root->window().dimensions(0).padding_low(), 0); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc455 padding.push_back(dim.padding_low()); in ImportInstructionImpl()
664 padding.push_back(dim.padding_low()); in ImportInstructionImpl()
697 paddings.push_back(dim.padding_low()); in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc308 window_dim.padding_low()); in CreateNaiveMlirConv()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc602 return static_cast<int64_t>(dim.padding_low()); in EmitSelectAndScatterOp()
750 return static_cast<int64_t>(dim.padding_low()); in EmitDnnConvolution()
1203 paddings.push_back(dim.padding_low()); in EmitReduceWindowOp()
Dlegalize_tf.cc3362 SmallVector<int64_t, 4> padding_low, padding_high, padding_interm; in matchAndRewrite() local
3364 padding_low.reserve(shape.size()); in matchAndRewrite()
3372 padding_low.push_back(begin_indices[i]); in matchAndRewrite()
3380 padding_low[i] + shape[i] + (shape[i] - 1) * padding_interm[i]; in matchAndRewrite()
3390 padding_low.push_back(input_dim - size); in matchAndRewrite()
3403 GetI64ElementsAttr(padding_low, &rewriter), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc1926 const auto& padding_low = op.edge_padding_low(); in Verify() local
1927 if (padding_low.getType().getNumElements() != input_type.getRank()) { in Verify()
1930 padding_low.getType().getNumElements(), input_type.getRank())); in Verify()
1957 int padding_low_val = padding_low.getValue<IntegerAttr>(i).getInt(); in Verify()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc184 padding.push_back(dim.padding_low()); in ReduceWindowInternal()

12