Searched refs:mirror_pad (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mirror_pad.cc | 38 namespace mirror_pad { namespace
289 static TfLiteRegistration r = {mirror_pad::Init, mirror_pad::Free, in Register_MIRROR_PAD()
290 mirror_pad::Prepare, mirror_pad::Eval}; in Register_MIRROR_PAD()
|
| D | Android.bp | 119 "mirror_pad.cc",
|
| D | BUILD | 596 "mirror_pad.cc",
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf-include-tf2xla-fallback.mlir | 29 // CHECK-LABEL: mirror_pad
30 func @mirror_pad(%arg0: tensor<2x3xcomplex<f64>>) -> tensor<4x7xcomplex<f64>> {
|
| D | legalize-tf-with-tf2xla.mlir | 226 // CHECK-LABEL: mirror_pad
227 func @mirror_pad(%arg0: tensor<2x3xcomplex<f64>>) -> tensor<4x7xcomplex<f64>> {
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | binary_ops_test.py | 1239 mirror_pad = lambda t, paddings: array_ops.pad(t, paddings, "SYMMETRIC") function
1242 mirror_pad,
1264 mirror_pad,
1270 mirror_pad = lambda t, paddings: array_ops.pad(t, paddings, "REFLECT") function
1273 mirror_pad,
1293 mirror_pad,
1298 mirror_pad,
1319 mirror_pad,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 217 Output mirror_pad = MirrorPad(root.WithOpName("mirror_pad"), casted_resize, in CompareFusedAndSeparate() local
219 Output conv = Conv2D(root.WithOpName("conv"), mirror_pad, casted_filter, in CompareFusedAndSeparate()
271 Output mirror_pad = MirrorPad(root.WithOpName("mirror_pad"), casted_input, in CompareFusedPadOnlyAndSeparate() local
273 Output conv = Conv2D(root.WithOpName("conv"), mirror_pad, casted_filter, in CompareFusedPadOnlyAndSeparate()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | ops_defs.py | 149 return [gen_array_ops.mirror_pad(grad, op.inputs[1], mode=mode), None]
|
| /external/tensorflow/tensorflow/lite/ |
| D | tflite_static.bp | 101 "kernels/mirror_pad.cc",
|
| D | build_def.bzl | 354 "mirror_pad",
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples_lib.py | 103 from tensorflow.lite.testing.op_tests.mirror_pad import make_mirror_pad_tests
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3519 result = gen_array_ops.mirror_pad(
3522 result = gen_array_ops.mirror_pad(
|
| D | array_grad.py | 973 return [gen_array_ops.mirror_pad(grad, op.inputs[1], mode=mode), None]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf.mlir | 1287 func @mirror_pad(tensor<2x1x3xf32>, tensor<3x2xi32>) -> tensor<? x f32> {
1292 // CHECK-LABEL: mirror_pad
1293 …// CHECK: "tfl.mirror_pad"(%arg0, %arg1) {mode = "SYMMETRIC"} : (tensor<2x1x3xf32>, tensor<3x2xi3…
1303 …// CHECK: "tfl.mirror_pad"(%arg0, %arg1) {mode = "REFLECT"} : (tensor<2x1x3xf32>, tensor<3x2xi32>…
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 1249 explicit PadOperationParser(bool mirror_pad) : mirror_pad_(mirror_pad) {} in PadOperationParser() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 3909 ### tfl.mirror_pad
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 3484 def TFL_MirrorPadOp: TFL_Op<"mirror_pad", [
|