Home
last modified time | relevance | path

Searched refs:padding_high (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc71 if (conv_window->dimensions(i).padding_high() > 0) { in MaybePaddedAndSlicedInput()
73 conv_window->dimensions(i).padding_high()); in MaybePaddedAndSlicedInput()
106 if (conv_window->dimensions(i).padding_high() < 0) { in MaybePaddedAndSlicedInput()
107 limit_indices[dim] -= -conv_window->dimensions(i).padding_high(); in MaybePaddedAndSlicedInput()
198 window_dim->set_padding_high(window_dim->padding_high() + delta); in IncreasePaddingHighBy()
227 int64 padding_high = backward_conv->window().dimensions(i).padding_high(); 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()
240 padding_high - new_conv_padding); in CanonicalizeBackwardFilterConvolution()
292 int64 padding_high = backward_conv->window().dimensions(i).padding_high(); in CanonicalizeBackwardInputConvolution() local
[all …]
Dgpu_conv_rewriter.cc290 if (dim->padding_high() < 0) { in MatchBackwardFilter()
294 << dim->padding_high() in MatchBackwardFilter()
489 if (dim->padding_high() < 0) { in MatchBackwardInput()
492 << dim->padding_high() in MatchBackwardInput()
Dgpu_conv_rewriter_test.cc319 EXPECT_EQ(3, window_dim.padding_high()); in TEST_F()
456 EXPECT_EQ(0, window_dim.padding_high()); in TEST_F()
560 EXPECT_EQ(1, backward_conv_col_dim.padding_high()); in TEST_F()
Dgpu_conv_runner.cc352 CHECK_EQ(dim.padding_low(), dim.padding_high()) << inst_as_string; in GetGpuConvConfig()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_pad_op.cc38 std::vector<int64> padding_high; in Compile() local
43 &padding_high)); in Compile()
55 OP_REQUIRES(context, rank == padding_high.size(), in Compile()
59 padding_high.size(), " vs. ", rank, ")")); in Compile()
76 dim->set_edge_padding_high(padding_high[i]); in Compile()
Dmatrix_diag_ops.cc205 int padding_high = 0; in SetMatrixDiag() local
208 std::swap(padding_low, padding_high); in SetMatrixDiag()
213 ->set_edge_padding_high(padding_high); in SetMatrixDiag()
/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.cc75 if (dim.padding_high() != 0) { in ToString()
76 StrAppend(&str, ",padding_high=", dim.padding_high()); 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()
220 window_dimension.padding_high() == 0 && in IsTrivialWindowDimension()
Dxla_data.proto441 int64 padding_high = 4; field
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_as_convolution_util.cc36 wd.padding_high() == 0 && wd.padding_low() == 0 && in ConvSpatialDimensionIsParallel()
42 if (wd.size() == lhs_size && wd.padding_high() == lhs_size - 1 && in ConvSpatialDimensionIsParallel()
77 wd.window_dilation() == 1 && wd.padding_high() == 0 && in ParseConvolutionDimsInfo()
85 if (rhs_size == 1 && wd.size() == 1 && wd.padding_high() == 0 && in ParseConvolutionDimsInfo()
91 wd.padding_high() == rhs_size - 1 && in ParseConvolutionDimsInfo()
Dconvolution_4d_expander.cc49 instruction->window().dimensions(i).padding_high() == 0) { in InstructionMatchesPattern()
80 instruction->window().dimensions(i).padding_high() == 0) { in ExpandInstruction()
Dspace_to_batch_converter.cc980 consumer->window().dimensions(old_space_dim).padding_high(); in CanPropagate()
1172 if (window.dimensions(old_space_dim).padding_high() > in SupportedOpForPropagation()
1453 consumer->window().dimensions(dim).padding_high()); in Propagate()
2409 .padding_high(); in PropagateOnBackpropFilterConv()
2614 .padding_high(); in GetConvolutionDetails()
Dhlo_cost_analysis.cc634 window_dim.padding_low() == 0 && window_dim.padding_high() == 0) { in HandleConvolution()
643 window_dim.padding_high() == output_limits[spatial_dimension] - 1 && in HandleConvolution()
Dalgebraic_simplifier_test.cc4413 EXPECT_EQ(root->window().dimensions(0).padding_high(), 100); in TEST_F()
4414 EXPECT_EQ(root->window().dimensions(1).padding_high(), 100); in TEST_F()
4415 EXPECT_EQ(root->window().dimensions(2).padding_high(), 100); in TEST_F()
4416 EXPECT_EQ(root->window().dimensions(3).padding_high(), 102); in TEST_F()
4499 EXPECT_EQ(root->window().dimensions(0).padding_high(), 100); in TEST_F()
4500 EXPECT_EQ(root->window().dimensions(1).padding_high(), 100); in TEST_F()
4501 EXPECT_EQ(root->window().dimensions(2).padding_high(), 100); in TEST_F()
4502 EXPECT_EQ(root->window().dimensions(3).padding_high(), 102); in TEST_F()
7151 EXPECT_EQ(conv->window().dimensions(0).padding_high(), 1); in TEST_F()
7152 EXPECT_EQ(conv->window().dimensions(1).padding_high(), 1); in TEST_F()
Ddynamic_padder.cc776 padding_dim->set_edge_padding_high(window_dim->padding_high() + in RewriteInputWithDynamicPadding()
791 window_dim->padding_high(); in RewriteInputWithDynamicPadding()
Dalgebraic_simplifier.cc4978 window_dim.set_padding_high(window_dim.padding_high() + in HandleReduceWindow()
5183 (w.padding_low() != 0 || w.padding_high() != 0 || in FoldConvInputPad()
5194 w.set_padding_high(w.padding_high() + p.edge_padding_high()); in FoldConvInputPad()
5300 window_dims[spatial_dim].padding_high() == 0 && in SwapConvOperands()
5345 window_dims[spatial_dim].padding_high() - in SwapConvOperands()
Dshape_inference.cc187 if (!allow_negative_padding && dim.padding_high() < 0) { in InferWindowOutputShape()
205 dim.padding_low() + dilated_base + dim.padding_high(); in InferWindowOutputShape()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py223 padding_high=[1, 2],
242 padding_high=[1, -2],
259 padding_high=[1, 2],
268 padding_high=[1, -2],
278 padding_high=[1, 1],
289 padding_high=[0, 0],
298 padding_high=[0, 0],
307 padding_high=[0, 0, 0],
316 padding_high=[0, 0],
326 padding_high=[0, 0],
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc374 int64 padding_high = wd.padding_high(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
376 int64 window_count = 1 + (padding_low + padding_high + base - in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
671 int64 padding_high = wd.padding_high(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
673 int64 window_count = 1 + (padding_low + padding_high + base - in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
931 wd->set_padding_high(new_window_dimension.padding_high()); in CreateShardedConvConvolution()
Dspmd_partitioner.cc595 wd.padding_high() + wd.padding_low(); in ReshardAsWindowedInput()
667 swd->padding_high(), in ReshardAsWindowedInput()
1618 limit_indices[i] = operand_shape.dimensions(i) + dim.padding_high(); in HandleSlice()
2656 dim->set_edge_padding_high(wd.padding_high()); in HandlePad()
2658 if (wd.padding_low() != 0 || wd.padding_high() != 0 || in HandlePad()
Dspmd_partitioner_test.cc1239 EXPECT_EQ(root->window().dimensions(0).padding_high(), 1); in TEST_F()
1299 EXPECT_EQ(root->operand(0)->window().dimensions(0).padding_high(), 0); in TEST_F()
1350 EXPECT_EQ(root->window().dimensions(0).padding_high(), 0); in TEST_F()
1404 EXPECT_EQ(root->window().dimensions(0).padding_high(), 0); in TEST_F()
1497 EXPECT_EQ(root->operand(0)->window().dimensions(0).padding_high(), 0); in TEST_F()
3422 EXPECT_EQ(root->window().dimensions(0).padding_high(), 0); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc456 padding.push_back(dim.padding_high()); in ImportInstructionImpl()
665 padding.push_back(dim.padding_high()); in ImportInstructionImpl()
698 paddings.push_back(dim.padding_high()); in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc1933 const auto& padding_high = op.edge_padding_high(); in Verify() local
1934 if (padding_high.getType().getNumElements() != input_type.getRank()) { in Verify()
1937 padding_high.getType().getNumElements(), input_type.getRank())); in Verify()
1958 int padding_high_val = padding_high.getValue<IntegerAttr>(i).getInt(); in Verify()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc185 padding.push_back(dim.padding_high()); in ReduceWindowInternal()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3362 SmallVector<int64_t, 4> padding_low, padding_high, padding_interm; in matchAndRewrite() local
3365 padding_high.reserve(shape.size()); in matchAndRewrite()
3381 padding_high.push_back(input_dim - size); in matchAndRewrite()
3384 padding_high.push_back(input_dim - begin_indices[i] - 1); in matchAndRewrite()
3389 padding_high[i] + shape[i] + (shape[i] - 1) * padding_interm[i]; in matchAndRewrite()
3404 GetI64ElementsAttr(padding_high, &rewriter), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc941 int64 padding_bottom = window.dimensions(0).padding_high(); in HandleConvolution()
945 one_dim_convolution ? 0 : window.dimensions(1).padding_high(); in HandleConvolution()

12