| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 543 int64 pad_low[3]; in PadArray3D() local
548 pad_low[i] = padding.dimensions(i).edge_padding_low(); in PadArray3D()
550 CHECK_LE(0, pad_low[i]); in PadArray3D()
556 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray3D()
569 bool in_low_padding = indices[i] < pad_low[i]; in PadArray3D()
576 (indices[i] - pad_low[i]) % (pad_interior[i] + 1)) { in PadArray3D()
584 *value = operand((indices[0] - pad_low[0]) / (pad_interior[0] + 1), in PadArray3D()
585 (indices[1] - pad_low[1]) / (pad_interior[1] + 1), in PadArray3D()
586 (indices[2] - pad_low[2]) / (pad_interior[2] + 1)); in PadArray3D()
602 int64 pad_low[4]; in PadArray4D() local
[all …]
|
| D | reference_util.cc | 158 int64 pad_low = padding[0].first; in ReduceWindow1DGeneric() local
163 int64 i0_base = i0 * stride_amount - pad_low; in ReduceWindow1DGeneric()
194 std::vector<int64> pad_low(window.size(), 0); in ReduceWindow3DAdd() local
198 pad_low[i] = padding_both[i].first; in ReduceWindow3DAdd()
206 int64 i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow3DAdd()
207 int64 i1_base = i1 * stride[1] - pad_low[1]; in ReduceWindow3DAdd()
208 int64 i2_base = i2 * stride[2] - pad_low[2]; in ReduceWindow3DAdd()
254 std::vector<int64> pad_low(window.size(), 0); in ReduceWindow4DGeneric() local
259 pad_low[i] = padding[i].first; in ReduceWindow4DGeneric()
268 int64 i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow4DGeneric()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 582 int64 pad_low[4]; member
597 "__pad_low_", absl::StrJoin(param.pad_low, "x"), // in R4ReduceWindowTestDataToString()
644 padding[i] = {param.pad_low[i], param.pad_high[i]}; in DoIt()
1077 int64 pad_low[2]; member
1221 "__pad_low_", absl::StrJoin(param.pad_low, "x"), "__pad_high_", in R2ReduceWindowTestDataToString()
1257 padding[i] = {param.pad_low[i], param.pad_high[i]}; in DoIt()
1292 int64 pad_low[1]; member
1426 "__pad_low_", absl::StrJoin(param.pad_low, "x"), in R1ReduceWindowTestDataToString()
1458 padding[0] = {param.pad_low[0], param.pad_high[0]}; in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 116 auto pad_low = padding.template getValue<llvm::APInt>({i, 0}); in MapWindowIvsToInput() local
119 Value pad_low_val = b->create<ConstantIndexOp>(loc, pad_low.getSExtValue()); in MapWindowIvsToInput()
|
| /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 | 1653 tensorflow::int64 pad_low; in inferConvReturnTypes() local
1657 pad_low = get_int(explicit_padding[2 * dim]); in inferConvReturnTypes()
1666 &pad_low, &pad_high); in inferConvReturnTypes()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgStructuredOps.td | 381 // x[0]*s[0] + d[0]*z[0] - pad_low[0],
383 // x[N-1]*s[N-1] + d[N-1]*z[N-1] - pad_low[N-1],
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 981 const int64 pad_low = in CanPropagate() local
985 pad_low) / in CanPropagate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3344 int64 pad_low = hlo->window().dimensions(i).padding_low(); in HandleSelectAndScatter() local
3347 if (data_left_halo_sizes[i].Calculate(0) == pad_low) { in HandleSelectAndScatter()
3354 LiteralUtil::CreateR0<int32>(pad_low))); in HandleSelectAndScatter()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1211 int64_t pad_low, pad_high; in matchAndRewrite() local
1213 pad_low = get_int(explicit_paddings[2 * dim]); in matchAndRewrite()
1223 pad_low = pad_low_int64; in matchAndRewrite()
1226 paddings.push_back(pad_low); in matchAndRewrite()
|