Searched refs:pad_high (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 544 int64 pad_high[3]; in PadArray3D() local
549 pad_high[i] = padding.dimensions(i).edge_padding_high(); in PadArray3D()
551 CHECK_LE(0, pad_high[i]); in PadArray3D()
556 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray3D()
570 bool in_high_padding = indices[i] >= output_bounds[i] - pad_high[i]; in PadArray3D()
603 int64 pad_high[4]; in PadArray4D() local
608 pad_high[i] = padding.dimensions(i).edge_padding_high(); in PadArray4D()
613 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray4D()
623 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 | 583 int64 pad_high[4]; member
598 "__pad_high_", absl::StrJoin(param.pad_high, "x"), // in R4ReduceWindowTestDataToString()
644 padding[i] = {param.pad_low[i], param.pad_high[i]}; in DoIt()
1078 int64 pad_high[2]; member
1222 absl::StrJoin(param.pad_high, "x"), "__layout_", param.layout[0], "_", in R2ReduceWindowTestDataToString()
1257 padding[i] = {param.pad_low[i], param.pad_high[i]}; in DoIt()
1293 int64 pad_high[1]; member
1427 "__pad_high_", absl::StrJoin(param.pad_high, "x"), in R1ReduceWindowTestDataToString()
1458 padding[0] = {param.pad_low[0], param.pad_high[0]}; in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 184 int64 pad_high = m - d; in SetMatrixDiagonal() local
187 pad_high = n - d; in SetMatrixDiagonal()
190 if (pad_high != 0) { in SetMatrixDiagonal()
201 dims->set_edge_padding_high(pad_high); in SetMatrixDiagonal()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo_patterns.td | 224 def : Pat<(HLO_PadOp:$old_value $input, $pad_value, $pad_low, $pad_high,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1654 tensorflow::int64 pad_high; in inferConvReturnTypes() local
1658 pad_high = get_int(explicit_padding[2 * dim + 1]); in inferConvReturnTypes()
1666 &pad_low, &pad_high); in inferConvReturnTypes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 979 const int64 pad_high = in CanPropagate() local
984 if ((first_operand->shape().dimensions(new_space_dim) + pad_high + in CanPropagate()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1211 int64_t pad_low, pad_high; in matchAndRewrite() local
1214 pad_high = get_int(explicit_paddings[2 * dim + 1]); in matchAndRewrite()
1224 pad_high = pad_high_int64; in matchAndRewrite()
1227 paddings.push_back(pad_high); in matchAndRewrite()
|