Home
last modified time | relevance | path

Searched refs:MirrorPadMode (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmirror_pad_op.cc32 const MirrorPadMode mode, in DoMirrorPad()
39 int64 excluded_edges = mode == MirrorPadMode::REFLECT ? 1 : 0; in DoMirrorPad()
68 MirrorPadMode mode; in Compile()
71 ctx, mode == MirrorPadMode::REFLECT || mode == MirrorPadMode::SYMMETRIC, in Compile()
/external/tensorflow/tensorflow/core/util/
Dmirror_pad_mode.cc25 MirrorPadMode* value) { in GetNodeAttr()
29 *value = MirrorPadMode::REFLECT; in GetNodeAttr()
31 *value = MirrorPadMode::SYMMETRIC; in GetNodeAttr()
Dmirror_pad_mode.h34 enum class MirrorPadMode { enum
48 MirrorPadMode* value);
/external/tensorflow/tensorflow/core/kernels/
Dmirror_pad_op.cc42 MirrorPadMode mode; in MirrorPadOp()
46 case MirrorPadMode::SYMMETRIC: { in MirrorPadOp()
50 case MirrorPadMode::REFLECT: { in MirrorPadOp()
253 MirrorPadMode mode; in MirrorPadGradOp()
257 case MirrorPadMode::SYMMETRIC: { in MirrorPadGradOp()
261 case MirrorPadMode::REFLECT: { in MirrorPadGradOp()
Dconv_ops_fused_image_transform.cc615 MirrorPadMode mode; in FusedResizeConv2DUsingGemmOp()
619 case MirrorPadMode::SYMMETRIC: { in FusedResizeConv2DUsingGemmOp()
623 case MirrorPadMode::REFLECT: { in FusedResizeConv2DUsingGemmOp()
/external/tensorflow/tensorflow/lite/experimental/writer/
Denum_mapping.h115 inline MirrorPadMode MirrorPaddingModeToSchema(TfLiteMirrorPaddingMode mode) { in MirrorPaddingModeToSchema()
/external/tensorflow/tensorflow/lite/kernels/
Dmirror_pad_test.cc32 const tflite::MirrorPadMode mode) { in BaseMirrorPadOpModel()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1875 *builder, op.mode == MirrorPadMode::kReflect in WriteOptions()
1876 ? ::tflite::MirrorPadMode::MirrorPadMode_REFLECT in WriteOptions()
1877 : ::tflite::MirrorPadMode::MirrorPadMode_SYMMETRIC); in WriteOptions()
1881 op->mode = options.mode() == ::tflite::MirrorPadMode::MirrorPadMode_REFLECT in ReadOptions()
1882 ? MirrorPadMode::kReflect in ReadOptions()
1883 : MirrorPadMode::kSymmetric; in ReadOptions()
Doperator_test.cc682 op.mode = MirrorPadMode::kReflect; in TEST_F()
/external/tensorflow/tensorflow/lite/schema/
Dschema.fbs728 enum MirrorPadMode : byte {
736 mode:MirrorPadMode;
Dschema_generated.h2367 enum MirrorPadMode {
2374 inline const MirrorPadMode (&EnumValuesMirrorPadMode())[2] {
2375 static const MirrorPadMode values[] = {
2391 inline const char *EnumNameMirrorPadMode(MirrorPadMode e) {
7240 MirrorPadMode mode;
7251 MirrorPadMode mode() const {
7252 return static_cast<MirrorPadMode>(GetField<int8_t>(VT_MODE, 0));
7267 void add_mode(MirrorPadMode mode) {
7284 MirrorPadMode mode = MirrorPadMode_REFLECT) {
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h2018 enum class MirrorPadMode { kNone, kSymmetric, kReflect };
2032 MirrorPadMode mode;
Dimport_tensorflow.cc1177 op->mode = toco::MirrorPadMode::kReflect; in ConvertMirrorPadOperator()
1179 op->mode = toco::MirrorPadMode::kSymmetric; in ConvertMirrorPadOperator()