| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | mirror_pad_op_gpu.cu.cc | 28 template struct functor::MirrorPad<GpuDevice, T, int32, 1>; \
29 template struct functor::MirrorPad<GpuDevice, T, int32, 2>; \
30 template struct functor::MirrorPad<GpuDevice, T, int32, 3>; \
31 template struct functor::MirrorPad<GpuDevice, T, int32, 4>; \
32 template struct functor::MirrorPad<GpuDevice, T, int32, 5>; \
33 template struct functor::MirrorPad<GpuDevice, T, int64, 1>; \
34 template struct functor::MirrorPad<GpuDevice, T, int64, 2>; \
35 template struct functor::MirrorPad<GpuDevice, T, int64, 3>; \
36 template struct functor::MirrorPad<GpuDevice, T, int64, 4>; \
37 template struct functor::MirrorPad<GpuDevice, T, int64, 5>; \
|
| D | mirror_pad_op_cpu_impl.h | 30 template struct functor::MirrorPad<CpuDevice, T, int32, CPU_PROVIDED_IXDIM>; \
31 template struct functor::MirrorPad<CpuDevice, T, int64, CPU_PROVIDED_IXDIM>;
|
| D | mirror_pad_op_benchmark_test.cc | 24 static Graph* MirrorPad(int batches, int height, int width, int depth, int pad, in MirrorPad() function
45 test::Benchmark(#DEVICE, MirrorPad(B, W, H, D, P, #MODE), \
|
| D | mirror_pad_op.cc | 124 functor::MirrorPad<Device, T, Tpaddings, i>()( \ in Compute()
157 void MirrorPad<CpuDevice, T, Tpaddings, i>::operator()( \
161 extern template struct MirrorPad<CpuDevice, T, Tpaddings, i>;
206 void MirrorPad<GpuDevice, T, Tpaddings, i>::operator()( \
210 extern template struct MirrorPad<GpuDevice, T, Tpaddings, i>;
|
| D | mirror_pad_op.h | 346 struct MirrorPad {
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MirrorPadGrad.pbtxt | 26 The mode used in the `MirrorPad` op.
29 summary: "Gradient op for `MirrorPad` op. This op folds a mirror-padded tensor."
31 This operation folds the padded areas of `input` by `MirrorPad` according to the
33 given to the corresponding `MirrorPad` op.
|
| D | api_def_MirrorPad.pbtxt | 2 graph_op_name: "MirrorPad"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf-include-tf2xla-fallback.mlir | 32 // NO_FALLBACK: tf.MirrorPad
33 // SUPPORTED_FALLBACK_DEVICE-NOT: tf.MirrorPad
34 // UNSPECIFIED_FALLBACK_DEVICE: tf.MirrorPad
35 // UNSUPPORTED_FALLBACK_DEVICE: tf.MirrorPad
36 …%1 = "tf.MirrorPad"(%arg0, %0) {mode = "SYMMETRIC"} : (tensor<2x3xcomplex<f64>>, tensor<2x2xi32>) …
|
| D | legalize-tf-with-tf2xla.mlir | 229 // CHECK-NOT: tf.MirrorPad
230 …%1 = "tf.MirrorPad"(%arg0, %0) {mode = "SYMMETRIC"} : (tensor<2x3xcomplex<f64>>, tensor<2x2xi32>) …
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_MirrorPad.pbtxt | 2 graph_op_name: "MirrorPad"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_MirrorPad.pbtxt | 2 graph_op_name: "MirrorPad"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MirrorPad.pbtxt | 2 name: "MirrorPad"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MirrorPad.pbtxt | 2 name: "MirrorPad"
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fuse_convolutions_test.cc | 64 MirrorPad(root.WithOpName("pad_op"), resize_op, pad_dims_op, "REFLECT"); in TestFuseResizePadAndConv()
166 MirrorPad(root.WithOpName("pad_op"), input_op, pad_dims_op, "REFLECT"); in TestFusePadAndConv()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | pad_ops_test.py | 56 self._assertOpAndComposite([input_], tf.raw_ops.MirrorPad,
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | README.md | 85 * MirrorPad
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 334 auto y = MirrorPad(scope_, x, paddings, "REFLECT"); in TEST_F()
343 auto y = MirrorPad(scope_, x, paddings, "SYMMETRIC"); in TEST_F()
|
| D | array_grad.cc | 409 grad_outputs->push_back(MirrorPad(scope, grad_inputs[0], op.input(1), mode)); in MirrorPadGradGrad()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | README.md | 82 * MirrorPad
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 217 Output mirror_pad = MirrorPad(root.WithOpName("mirror_pad"), casted_resize, in CompareFusedAndSeparate()
271 Output mirror_pad = MirrorPad(root.WithOpName("mirror_pad"), casted_input, in CompareFusedPadOnlyAndSeparate()
|
| /external/tensorflow/tensorflow/lite/g3doc/performance/ |
| D | coreml_delegate.md | 259 * MirrorPad
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 279 * `raw_ops.MirrorPad`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 146 `MirrorPad` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
|
| D | cpu_supported_ops.md | 146 `MirrorPad` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1494 class MirrorPad class
1966 ops.push_back(MakeUnique<MirrorPad>(::tflite::BuiltinOperator_MIRROR_PAD, in BuildOperatorList()
|