| /external/tensorflow/tensorflow/core/kernels/ |
| mirror_pad_op_gpu.cu.cc | 29 template struct functor::MirrorPad<GpuDevice, T, int32, 1>; \
30 template struct functor::MirrorPad<GpuDevice, T, int32, 2>; \
31 template struct functor::MirrorPad<GpuDevice, T, int32, 3>; \
32 template struct functor::MirrorPad<GpuDevice, T, int32, 4>; \
33 template struct functor::MirrorPad<GpuDevice, T, int32, 5>; \
34 template struct functor::MirrorPad<GpuDevice, T, int64, 1>; \
35 template struct functor::MirrorPad<GpuDevice, T, int64, 2>; \
36 template struct functor::MirrorPad<GpuDevice, T, int64, 3>; \
37 template struct functor::MirrorPad<GpuDevice, T, int64, 4>; \
38 template struct functor::MirrorPad<GpuDevice, T, int64, 5>;
[all...] |
| mirror_pad_op_cpu_impl.h | 29 template struct functor::MirrorPad<CpuDevice, T, int32, CPU_PROVIDED_IXDIM>; \
30 template struct functor::MirrorPad<CpuDevice, T, int64, CPU_PROVIDED_IXDIM>;
|
| mirror_pad_op.cc | 124 functor::MirrorPad<Device, T, Tpaddings, i>()( \
157 void MirrorPad<CpuDevice, T, Tpaddings, i>::operator()( \
161 extern template struct MirrorPad<CpuDevice, T, Tpaddings, i>;
183 REGISTER_KERNEL_BUILDER(Name("MirrorPad") \
189 REGISTER_KERNEL_BUILDER(Name("MirrorPad") \
206 void MirrorPad<GpuDevice, T, Tpaddings, i>::operator()( \
210 extern template struct MirrorPad<GpuDevice, T, Tpaddings, i>;
231 REGISTER_KERNEL_BUILDER(Name("MirrorPad") \
237 REGISTER_KERNEL_BUILDER(Name("MirrorPad") \
|
| mirror_pad_op.h | 340 struct MirrorPad {
|
| conv_ops_test.cc | 214 Output mirror_pad = MirrorPad(root.WithOpName("mirror_pad"), casted_resize,
268 Output mirror_pad = MirrorPad(root.WithOpName("mirror_pad"), casted_input,
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fuse_convolutions_test.cc | 64 MirrorPad(root.WithOpName("pad_op"), resize_op, pad_dims_op, "REFLECT");
96 EXPECT_NE("MirrorPad", node.op());
166 MirrorPad(root.WithOpName("pad_op"), input_op, pad_dims_op, "REFLECT");
198 EXPECT_NE("MirrorPad", node.op());
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| array_grad_test.cc | 334 auto y = MirrorPad(scope_, x, paddings, "REFLECT");
343 auto y = MirrorPad(scope_, x, paddings, "SYMMETRIC");
|
| array_grad.cc | 392 REGISTER_GRADIENT_OP("MirrorPad", MirrorPadGrad);
400 grad_outputs->push_back(MirrorPad(scope, grad_inputs[0], op.input(1), mode));
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| operator.cc | [all...] |
| /external/tensorflow/tensorflow/lite/toco/ |
| tooling_util.cc | 423 HANDLE_OPERATORTYPENAME_CASE(MirrorPad)
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
