Searched refs:padding_interior (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_pad_op.cc | 39 std::vector<int64> padding_interior; in Compile() local
45 "padding_interior", &padding_interior)); in Compile()
60 OP_REQUIRES(context, rank == padding_interior.size(), in Compile()
64 padding_interior.size(), " vs. ", rank, ")")); in Compile()
68 context, absl::c_all_of(padding_interior, non_negative), in Compile()
70 absl::StrJoin(padding_interior, ","), "]")); in Compile()
77 dim->set_interior_padding(padding_interior[i]); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 224 padding_interior=[1, 0])
243 padding_interior=[1, 2])
260 padding_interior=[1, 4])
269 padding_interior=[1, 2])
279 padding_interior=[1, 2])
290 padding_interior=[0, 0])
299 padding_interior=[0, 0])
308 padding_interior=[0, 0])
317 padding_interior=[0])
327 padding_interior=[-2, 0])
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1940 const auto& padding_interior = op.interior_padding(); in Verify() local
1941 if (padding_interior.getType().getNumElements() != input_type.getRank()) { in Verify()
1944 padding_interior.getType().getNumElements(), input_type.getRank())); in Verify()
1960 padding_interior.getValue<IntegerAttr>(i).getInt(); in Verify()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 7687 …ing_value tf.Output, padding_low tf.Output, padding_high tf.Output, padding_interior tf.Output) (o…
7694 input, padding_value, padding_low, padding_high, padding_interior,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 17882 containing only non-negative values.}]>:$padding_interior
|