Home
last modified time | relevance | path

Searched refs:pad_low (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc166 std::vector<int64> pad_low(window.size(), 0); in ReduceWindow1DGeneric() local
171 pad_low[i] = padding[i].first; in ReduceWindow1DGeneric()
177 int64 i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow1DGeneric()
211 std::vector<int64> pad_low(window.size(), 0); in ReduceWindow2DGeneric() local
216 pad_low[i] = padding[i].first; in ReduceWindow2DGeneric()
224 int64 i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow2DGeneric()
225 int64 i1_base = i1 * stride[1] - pad_low[1]; in ReduceWindow2DGeneric()
260 std::vector<int64> pad_low(window.size(), 0); in ReduceWindow3DAdd() local
264 pad_low[i] = padding_both[i].first; in ReduceWindow3DAdd()
272 int64 i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow3DAdd()
[all …]
Dreference_util.h551 int64 pad_low[3]; in PadArray3D() local
556 pad_low[i] = padding.dimensions(i).edge_padding_low(); in PadArray3D()
558 CHECK_LE(0, pad_low[i]); in PadArray3D()
564 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray3D()
577 bool in_low_padding = indices[i] < pad_low[i]; in PadArray3D()
584 (indices[i] - pad_low[i]) % (pad_interior[i] + 1)) { in PadArray3D()
592 *value = operand((indices[0] - pad_low[0]) / (pad_interior[0] + 1), in PadArray3D()
593 (indices[1] - pad_low[1]) / (pad_interior[1] + 1), in PadArray3D()
594 (indices[2] - pad_low[2]) / (pad_interior[2] + 1)); in PadArray3D()
610 int64 pad_low[4]; in PadArray4D() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc567 int64 pad_low[4]; member
582 "__pad_low_", absl::StrJoin(param.pad_low, "x"), // in R4ReduceWindowTestDataToString()
628 padding[i] = {param.pad_low[i], param.pad_high[i]}; in DoIt()
1059 int64 pad_low[2]; member
1141 "__pad_low_", absl::StrJoin(param.pad_low, "x"), "__pad_high_", in R2ReduceWindowTestDataToString()
1171 padding[i] = {param.pad_low[i], param.pad_high[i]}; in DoIt()
1213 int64 pad_low[1]; member
1333 "__pad_low_", absl::StrJoin(param.pad_low, "x"), in R1ReduceWindowTestDataToString()
1364 padding[0] = {param.pad_low[0], param.pad_high[0]}; in TEST_P()