Searched refs:pad_high (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 552 int64 pad_high[3]; in PadArray3D() local
557 pad_high[i] = padding.dimensions(i).edge_padding_high(); in PadArray3D()
559 CHECK_LE(0, pad_high[i]); in PadArray3D()
564 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray3D()
578 bool in_high_padding = indices[i] >= output_bounds[i] - pad_high[i]; in PadArray3D()
611 int64 pad_high[4]; in PadArray4D() local
616 pad_high[i] = padding.dimensions(i).edge_padding_high(); in PadArray4D()
621 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray4D()
631 bool in_high_padding = indices[i] >= output_bounds[i] - pad_high[i]; in PadArray4D()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 568 int64 pad_high[4]; member
583 "__pad_high_", absl::StrJoin(param.pad_high, "x"), // in R4ReduceWindowTestDataToString()
628 padding[i] = {param.pad_low[i], param.pad_high[i]}; in DoIt()
1060 int64 pad_high[2]; member
1142 absl::StrJoin(param.pad_high, "x"), "__layout_", param.layout[0], "_", in R2ReduceWindowTestDataToString()
1171 padding[i] = {param.pad_low[i], param.pad_high[i]}; in DoIt()
1214 int64 pad_high[1]; member
1334 "__pad_high_", absl::StrJoin(param.pad_high, "x"), in R1ReduceWindowTestDataToString()
1364 padding[0] = {param.pad_low[0], param.pad_high[0]}; in TEST_P()
|