Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_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 …]
Dcudnn_conv_rewriter.cc212 if (dim->padding_high() < 0) { in MatchBackwardFilter()
216 << dim->padding_high() in MatchBackwardFilter()
419 if (dim->padding_high() < 0) { in MatchBackwardInput()
422 << dim->padding_high() in MatchBackwardInput()
Dcudnn_conv_rewriter_test.cc316 EXPECT_EQ(3, window_dim.padding_high()); in TEST_F()
449 EXPECT_EQ(0, window_dim.padding_high()); in TEST_F()
551 EXPECT_EQ(1, backward_conv_col_dim.padding_high()); in TEST_F()
Dcudnn_conv_runner.cc161 CHECK_EQ(dim.padding_low(), dim.padding_high()); in RunCudnnConvImpl()
/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()
72 context, absl::c_all_of(padding_high, non_negative), in Compile()
74 absl::StrJoin(padding_high, ","), "]")); in Compile()
84 dim->set_edge_padding_high(padding_high[i]); in Compile()
/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.cc61 if (dim.padding_high() != 0) { in ToString()
62 StrAppend(&str, ",padding_high=", dim.padding_high()); in ToString()
101 return StrCat(dim.padding_low(), "_", dim.padding_high()); in ToString()
133 if (dim.padding_low() != 0 || dim.padding_high() != 0) { in HasPadding()
142 return dim.padding_low() == dim.padding_high(); in HasSymmetricPadding()
156 return dim.padding_low() < 0 || dim.padding_high() < 0; in HasNegativePadding()
204 window_dim.padding_low() == 0 && window_dim.padding_high() == 0; in IsInactiveWindowDimension()
210 window_dimension.padding_high() == 0 && in IsTrivialWindowDimension()
Dxla_data.proto406 int64 padding_high = 4; field
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py198 padding_high=[1, 2],
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc3645 EXPECT_EQ(root->window().dimensions(0).padding_high(), 100); in TEST_F()
3646 EXPECT_EQ(root->window().dimensions(1).padding_high(), 100); in TEST_F()
3647 EXPECT_EQ(root->window().dimensions(2).padding_high(), 100); in TEST_F()
3648 EXPECT_EQ(root->window().dimensions(3).padding_high(), 102); in TEST_F()
3731 EXPECT_EQ(root->window().dimensions(0).padding_high(), 100); in TEST_F()
3732 EXPECT_EQ(root->window().dimensions(1).padding_high(), 100); in TEST_F()
3733 EXPECT_EQ(root->window().dimensions(2).padding_high(), 100); in TEST_F()
3734 EXPECT_EQ(root->window().dimensions(3).padding_high(), 102); in TEST_F()
Dalgebraic_simplifier.cc3295 window_dim.set_padding_high(window_dim.padding_high() + in HandleReduceWindow()
3434 (w.padding_low() != 0 || w.padding_high() != 0 || in FoldConvInputPad()
3445 w.set_padding_high(w.padding_high() + p.edge_padding_high()); in FoldConvInputPad()
Dshape_inference.cc185 if (!allow_negative_padding && dim.padding_high() < 0) { in InferWindowOutputShape()
210 dim.padding_low() + dilated_base + dim.padding_high(); in InferWindowOutputShape()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1164 int64 padding_bottom = window.dimensions(0).padding_high(); in HandleConvolution()
1168 one_dim_convolution ? 0 : window.dimensions(1).padding_high(); in HandleConvolution()