Home
last modified time | relevance | path

Searched refs:MirrorPad (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmirror_pad_op_gpu.cu.cc29 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>; \
Dmirror_pad_op_cpu_impl.h29 template struct functor::MirrorPad<CpuDevice, T, int32, CPU_PROVIDED_IXDIM>; \
30 template struct functor::MirrorPad<CpuDevice, T, int64, CPU_PROVIDED_IXDIM>;
Dmirror_pad_op.cc124 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>;
Dmirror_pad_op.h340 struct MirrorPad {
Dconv_ops_test.cc214 Output mirror_pad = MirrorPad(root.WithOpName("mirror_pad"), casted_resize, in CompareFusedAndSeparate()
268 Output mirror_pad = MirrorPad(root.WithOpName("mirror_pad"), casted_input, in CompareFusedPadOnlyAndSeparate()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_MirrorPadGrad.pbtxt26 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.
Dapi_def_MirrorPad.pbtxt2 graph_op_name: "MirrorPad"
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_MirrorPad.pbtxt2 graph_op_name: "MirrorPad"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_MirrorPad.pbtxt2 graph_op_name: "MirrorPad"
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfuse_convolutions_test.cc64 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()
DREADME.md515 For graphs that use ResizeBilinear or MirrorPad ops before convolutions (e.g. to
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad_test.cc334 auto y = MirrorPad(scope_, x, paddings, "REFLECT"); in TEST_F()
343 auto y = MirrorPad(scope_, x, paddings, "SYMMETRIC"); in TEST_F()
Darray_grad.cc400 grad_outputs->push_back(MirrorPad(scope, grad_inputs[0], op.input(1), mode)); in MirrorPadGradGrad()
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md146 `MirrorPad` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
Dcpu_supported_ops.md146 `MirrorPad` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1866 class MirrorPad class
2467 ops.push_back(MakeUnique<MirrorPad>(::tflite::BuiltinOperator_MIRROR_PAD, in BuildOperatorList()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc423 HANDLE_OPERATORTYPENAME_CASE(MirrorPad) in OperatorTypeName()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt2032 name: "MirrorPad"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt2032 name: "MirrorPad"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt18013 name: "MirrorPad"
18042 name: "MirrorPad"
Dops_history.v1.pbtxt40528 name: "MirrorPad"
Dops_history.v2.pbtxt40090 name: "MirrorPad"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go1252 func MirrorPad(scope *Scope, input tf.Output, paddings tf.Output, mode string) (output tf.Output) { func
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt19825 name: "MirrorPad"