Searched refs:paddings_value (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pad_op_test.py | 387 for paddings_value in paddings_values:
390 paddings = constant_op.constant(paddings_value, dtype=dtypes.int32)
392 middle = array_ops.slice(padded, [row[0] for row in paddings_value],
395 [row[0] for row in paddings_value])
398 [paddings_value[i][0] + inp.shape.dims[i].value for i in range(4)],
403 np.zeros([row[0] for row in paddings_value]), self.evaluate(left))
405 np.zeros([row[1] for row in paddings_value]),
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 83 Value paddings_value);
|
| D | legalize_common.cc | 735 Value paddings_value) { in convertSpaceToBatchNDOp() argument
796 paddings_value.getType().dyn_cast<RankedTensorType>(); in convertSpaceToBatchNDOp()
839 if (!matchPattern(paddings_value, m_Constant(&paddings_elems))) in convertSpaceToBatchNDOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 226 auto paddings_value = builder->create<TF::ConstOp>(filter.getLoc(), paddings); in GetPadOpForConv2DFilter() local
235 paddings_value); in GetPadOpForConv2DFilter()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 314 auto paddings_value = paddings_op.value().dyn_cast<DenseElementsAttr>(); in FoldOperandsPermutation() local
315 if (!paddings_value || in FoldOperandsPermutation()
316 paddings_value.getNumElements() != permutation.size() * 2) in FoldOperandsPermutation()
319 SmallVector<int32_t, 8> shuffled_paddings(paddings_value.getNumElements()); in FoldOperandsPermutation()
320 for (auto index_pair : llvm::enumerate(paddings_value.getIntValues())) { in FoldOperandsPermutation()
330 auto type = mlir::RankedTensorType::get(paddings_value.getType().getShape(), in FoldOperandsPermutation()
|