Home
last modified time | relevance | path

Searched refs:MirrorPadMode (Results 1 – 14 of 14) 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/compiler/mlir/lite/
Dflatbuffer_operator.cc89 static tflite::MirrorPadMode ConvertTFL_MirrorPaddingAttrForOptionWriter( in ConvertTFL_MirrorPaddingAttrForOptionWriter()
91 return llvm::StringSwitch<tflite::MirrorPadMode>(str) in ConvertTFL_MirrorPaddingAttrForOptionWriter()
233 static mlir::Attribute BuildTFL_MirrorPaddingAttr(tflite::MirrorPadMode value, in BuildTFL_MirrorPaddingAttr()
/external/tensorflow/tensorflow/lite/experimental/writer/
Denum_mapping.h119 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.cc1423 *builder, op.mode == MirrorPadMode::kReflect in WriteOptions()
1424 ? ::tflite::MirrorPadMode::MirrorPadMode_REFLECT in WriteOptions()
1425 : ::tflite::MirrorPadMode::MirrorPadMode_SYMMETRIC); in WriteOptions()
1429 op->mode = options.mode() == ::tflite::MirrorPadMode::MirrorPadMode_REFLECT in ReadOptions()
1430 ? MirrorPadMode::kReflect in ReadOptions()
1431 : MirrorPadMode::kSymmetric; in ReadOptions()
Doperator_test.cc727 op.mode = MirrorPadMode::kReflect; in TEST_F()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h2077 enum class MirrorPadMode { kNone, kSymmetric, kReflect };
2091 MirrorPadMode mode;
Dimport_tensorflow.cc1392 op->mode = toco::MirrorPadMode::kReflect; in ConvertMirrorPadOperator()
1394 op->mode = toco::MirrorPadMode::kSymmetric; in ConvertMirrorPadOperator()
/external/tensorflow/tensorflow/lite/schema/
Dschema.fbs855 enum MirrorPadMode : byte {
863 mode:MirrorPadMode;
Dschema_generated.h2693 enum MirrorPadMode {
2700 inline const MirrorPadMode (&EnumValuesMirrorPadMode())[2] {
2701 static const MirrorPadMode values[] = {
2717 inline const char *EnumNameMirrorPadMode(MirrorPadMode e) {
7926 MirrorPadMode mode;
7937 MirrorPadMode mode() const {
7938 return static_cast<MirrorPadMode>(GetField<int8_t>(VT_MODE, 0));
7953 void add_mode(MirrorPadMode mode) {
7970 MirrorPadMode mode = MirrorPadMode_REFLECT) {