Searched refs:MirrorPadOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | mirror_pad_op.cc | 25 class MirrorPadOp : public XlaOpKernel { class
27 explicit MirrorPadOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in MirrorPadOp() function in tensorflow::__anond15c4dac0111::MirrorPadOp
105 TF_DISALLOW_COPY_AND_ASSIGN(MirrorPadOp);
109 MirrorPadOp);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mirror_pad_op.cc | 39 class MirrorPadOp : public OpKernel { class
41 explicit MirrorPadOp(OpKernelConstruction* context) : OpKernel(context) { in MirrorPadOp() function in tensorflow::MirrorPadOp
61 ~MirrorPadOp() override = default;
188 MirrorPadOp<CpuDevice, type, int32>); \
194 MirrorPadOp<CpuDevice, type, int64>);
236 MirrorPadOp<GpuDevice, T, int32>); \
242 MirrorPadOp<GpuDevice, T, int64>);
|
| D | mirror_pad_op.h | 351 output.device(device) = MirrorPadOp(input, padding_dims, offset);
356 MirrorPadOp(
|