Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_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 …]
Dcudnn_conv_rewriter.cc141 if (window_dim.padding_low() < 0) { in MatchBackwardFilter()
173 dim->set_padding_low(conv->window().dimensions(i).padding_low()); in MatchBackwardFilter()
192 padded_input_size - input_size - dim->padding_low(); in MatchBackwardFilter()
194 CHECK_GE(dim->padding_low(), 0); in MatchBackwardFilter()
202 if (dim->padding_low() >= min_padding_high && in MatchBackwardFilter()
203 dim->padding_low() <= max_padding_high) { in MatchBackwardFilter()
204 dim->set_padding_high(dim->padding_low()); in MatchBackwardFilter()
206 if (dim->padding_low() < min_padding_high) { in MatchBackwardFilter()
350 kernel_size - 1 - old_window.dimensions(i).padding_low(); in MatchBackwardInput()
Dcudnn_conv_rewriter_test.cc315 EXPECT_EQ(3, window_dim.padding_low()); in TEST_F()
448 EXPECT_EQ(0, window_dim.padding_low()); in TEST_F()
550 EXPECT_EQ(0, backward_conv_col_dim.padding_low()); in TEST_F()
Dcudnn_conv_runner.cc161 CHECK_EQ(dim.padding_low(), dim.padding_high()); in RunCudnnConvImpl()
212 window.dimensions(dim).padding_low()) in RunCudnnConvImpl()
Delemental_ir_emitter.cc398 index_typed_const(window.dimensions(i).padding_low())); in MakeElementGenerator()
Dir_emitter_unnested.cc873 index_typed_constant(window.dimensions(i).padding_low())); in HandleSelectAndScatter()
/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()
68 context, absl::c_all_of(padding_low, non_negative), in Compile()
70 absl::StrJoin(padding_low, ","), "]")); in Compile()
83 dim->set_edge_padding_low(padding_low[i]); in Compile()
/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.cc58 if (dim.padding_low() != 0) { in ToString()
59 StrAppend(&str, ",padding_low=", dim.padding_low()); 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()
209 window_dimension.padding_low() == 0 && in IsTrivialWindowDimension()
Dxla_data.proto401 int64 padding_low = 3; field
403 // As padding_low, but on the high end of this dimension. For example, in the
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py197 padding_low=[2, 1],
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc677 b_.getInt64(window.dimensions(i).padding_low())); in EmitElementalReduceWindow()
829 b_.getInt64(window.dimensions(i).padding_low())); in HandleSelectAndScatter()
1009 b_.getInt64(window_dim.padding_low())); in EmitElementalConvolution()
1163 int64 padding_top = window.dimensions(0).padding_low(); in HandleConvolution()
1166 one_dim_convolution ? 0 : window.dimensions(1).padding_low(); in HandleConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc3641 EXPECT_EQ(root->window().dimensions(0).padding_low(), 10); in TEST_F()
3642 EXPECT_EQ(root->window().dimensions(1).padding_low(), 11); in TEST_F()
3643 EXPECT_EQ(root->window().dimensions(2).padding_low(), 10); in TEST_F()
3644 EXPECT_EQ(root->window().dimensions(3).padding_low(), 10); in TEST_F()
3727 EXPECT_EQ(root->window().dimensions(0).padding_low(), 10); in TEST_F()
3728 EXPECT_EQ(root->window().dimensions(1).padding_low(), 11); in TEST_F()
3729 EXPECT_EQ(root->window().dimensions(2).padding_low(), 10); in TEST_F()
3730 EXPECT_EQ(root->window().dimensions(3).padding_low(), 10); in TEST_F()
Dhlo_cost_analysis.cc500 window_dim.padding_low() + in HandleConvolution()
Dalgebraic_simplifier.cc3293 window_dim.set_padding_low(window_dim.padding_low() + in HandleReduceWindow()
3434 (w.padding_low() != 0 || w.padding_high() != 0 || in FoldConvInputPad()
3444 w.set_padding_low(w.padding_low() + p.edge_padding_low()); in FoldConvInputPad()
Dshape_inference.cc181 if (!allow_negative_padding && dim.padding_low() < 0) { in InferWindowOutputShape()
210 dim.padding_low() + dilated_base + dim.padding_high(); in InferWindowOutputShape()
Dhlo_evaluator_typed_visitor.h1126 window_dim.padding_low() + in HandleConvolution()
2837 window.dimensions(i).padding_low(); in IterateThroughWindow()