| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 556 int64_t pad_low[3]; in PadArray3D() local
561 pad_low[i] = padding.dimensions(i).edge_padding_low(); in PadArray3D()
563 CHECK_LE(0, pad_low[i]); in PadArray3D()
569 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray3D()
582 bool in_low_padding = indices[i] < pad_low[i]; in PadArray3D()
589 (indices[i] - pad_low[i]) % (pad_interior[i] + 1)) { in PadArray3D()
597 *value = operand((indices[0] - pad_low[0]) / (pad_interior[0] + 1), in PadArray3D()
598 (indices[1] - pad_low[1]) / (pad_interior[1] + 1), in PadArray3D()
599 (indices[2] - pad_low[2]) / (pad_interior[2] + 1)); in PadArray3D()
615 int64_t pad_low[4]; in PadArray4D() local
[all …]
|
| D | reference_util.cc | 159 int64_t pad_low = padding[0].first; in ReduceWindow1DGeneric() local
164 int64_t i0_base = i0 * stride_amount - pad_low; in ReduceWindow1DGeneric()
195 std::vector<int64_t> pad_low(window.size(), 0); in ReduceWindow3DAdd() local
199 pad_low[i] = padding_both[i].first; in ReduceWindow3DAdd()
207 int64_t i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow3DAdd()
208 int64_t i1_base = i1 * stride[1] - pad_low[1]; in ReduceWindow3DAdd()
209 int64_t i2_base = i2 * stride[2] - pad_low[2]; in ReduceWindow3DAdd()
255 std::vector<int64_t> pad_low(window.size(), 0); in ReduceWindow4DGeneric() local
261 pad_low[i] = padding[i].first; in ReduceWindow4DGeneric()
270 int64_t i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow4DGeneric()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 580 int64_t pad_low[4]; member
595 "__pad_low_", absl::StrJoin(param.pad_low, "x"), // in R4ReduceWindowTestDataToString()
642 padding[i] = {param.pad_low[i], param.pad_high[i]}; in DoIt()
1102 int64_t pad_low[2]; member
1267 "__pad_low_", absl::StrJoin(param.pad_low, "x"), "__pad_high_", in R2ReduceWindowTestDataToString()
1306 padding[i] = {param.pad_low[i], param.pad_high[i]}; in DoIt()
1341 int64_t pad_low[1]; member
1496 "__pad_low_", absl::StrJoin(param.pad_low, "x"), in R1ReduceWindowTestDataToString()
1531 padding[0] = {param.pad_low[0], param.pad_high[0]}; in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1777 int64_t pad_low; in inferConvReturnTypeComponents() local
1781 pad_low = get_int(explicit_padding[2 * dim]); in inferConvReturnTypeComponents()
1790 &pad_low, &pad_high); in inferConvReturnTypeComponents()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1181 Value pad_low, pad_high; in matchAndRewriteDynamicConv() local
1183 pad_low = get_const(get_int(explicit_paddings[2 * dim])); in matchAndRewriteDynamicConv()
1189 stride, padding, shape_scalar_type, &pad_low, in matchAndRewriteDynamicConv()
1194 paddings.push_back(pad_low); in matchAndRewriteDynamicConv()
1331 int64_t pad_low, pad_high; in matchAndRewrite() local
1333 pad_low = get_int(explicit_paddings[2 * dim]); in matchAndRewrite()
1345 pad_low = pad_low_int64; in matchAndRewrite()
1348 paddings.push_back(pad_low); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo_patterns.td | 234 def : Pat<(HLO_PadOp:$old_value $input, $pad_value, $pad_low, $pad_high,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1192 int64_t pad_low; in ComputeConvWindowedOutputSize() local
1197 &pad_low, &pad_high); in ComputeConvWindowedOutputSize()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 1389 const int64_t pad_low = in CanPropagate() local
1393 pad_low) / in CanPropagate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 4153 int64_t pad_low = hlo->window().dimensions(i).padding_low(); in HandleSelectAndScatter() local
4156 if (data_left_halo_sizes[i].Calculate(0) == pad_low) { in HandleSelectAndScatter()
4163 LiteralUtil::CreateR0<int32_t>(pad_low))); in HandleSelectAndScatter()
|