Searched refs:MirrorPadGrad (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | mirror_pad_op_gpu.cu.cc | 38 template struct functor::MirrorPadGrad<GpuDevice, T, int32, 1>; \
39 template struct functor::MirrorPadGrad<GpuDevice, T, int32, 2>; \
40 template struct functor::MirrorPadGrad<GpuDevice, T, int32, 3>; \
41 template struct functor::MirrorPadGrad<GpuDevice, T, int32, 4>; \
42 template struct functor::MirrorPadGrad<GpuDevice, T, int32, 5>; \
43 template struct functor::MirrorPadGrad<GpuDevice, T, int64, 1>; \
44 template struct functor::MirrorPadGrad<GpuDevice, T, int64, 2>; \
45 template struct functor::MirrorPadGrad<GpuDevice, T, int64, 3>; \
46 template struct functor::MirrorPadGrad<GpuDevice, T, int64, 4>; \
47 template struct functor::MirrorPadGrad<GpuDevice, T, int64, 5>;
|
| D | mirror_pad_op_cpu_impl.h | 37 template struct functor::MirrorPadGrad<CpuDevice, T, int32, \
39 template struct functor::MirrorPadGrad<CpuDevice, T, int64, \
|
| D | mirror_pad_op.cc | 335 functor::MirrorPadGrad<Device, T, Tpaddings, k>()( \ in Compute()
366 void MirrorPadGrad<CpuDevice, T, Tpaddings, k>::operator()( \
371 extern template struct MirrorPadGrad<CpuDevice, T, Tpaddings, k>;
412 void MirrorPadGrad<GpuDevice, T, Tpaddings, k>::operator()( \
417 extern template struct MirrorPadGrad<GpuDevice, T, Tpaddings, k>;
|
| D | mirror_pad_op.h | 373 struct MirrorPadGrad {
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_MirrorPadGrad.pbtxt | 2 graph_op_name: "MirrorPadGrad"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MirrorPadGrad.pbtxt | 2 name: "MirrorPadGrad"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MirrorPadGrad.pbtxt | 2 name: "MirrorPadGrad"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MirrorPadGrad.pbtxt | 2 graph_op_name: "MirrorPadGrad"
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | pad_ops_test.py | 84 self._assertOpAndComposite([input_], tf.raw_ops.MirrorPadGrad,
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 29 using ops::internal::MirrorPadGrad;
352 auto y = MirrorPadGrad(scope_, x, paddings, "REFLECT"); in TEST_F()
361 auto y = MirrorPadGrad(scope_, x, paddings, "SYMMETRIC"); in TEST_F()
|
| D | array_grad.cc | 391 Status MirrorPadGrad(const Scope& scope, const Operation& op, in MirrorPadGrad() function
396 grad_outputs->push_back(tensorflow::ops::internal::MirrorPadGrad( in MirrorPadGrad()
401 REGISTER_GRADIENT_OP("MirrorPad", MirrorPadGrad);
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 280 * `raw_ops.MirrorPadGrad`
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2536 name: "MirrorPadGrad"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2536 name: "MirrorPadGrad"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 18084 name: "MirrorPadGrad"
18113 name: "MirrorPadGrad"
|
| D | ops_history.v2.pbtxt | 39469 name: "MirrorPadGrad"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 46276 func MirrorPadGrad(scope *Scope, input tf.Output, paddings tf.Output, mode string) (output tf.Outpu… func
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 8439 def TF_MirrorPadGradOp : TF_Op<"MirrorPadGrad", [NoSideEffect, TF_OperandHasRank<1, 2>]> {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 25350 name: "MirrorPadGrad"
|