Searched refs:MirrorPadMode (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | mirror_pad_op.cc | 34 const MirrorPadMode mode, in DoMirrorPad()
41 int64 excluded_edges = mode == MirrorPadMode::REFLECT ? 1 : 0; in DoMirrorPad()
70 MirrorPadMode mode; in Compile()
73 ctx, mode == MirrorPadMode::REFLECT || mode == MirrorPadMode::SYMMETRIC, in Compile()
120 const xla::LiteralSlice& pad_literal, const MirrorPadMode mode, in DoMirrorPadGrad()
127 int64 excluded_edges = mode == MirrorPadMode::REFLECT ? 1 : 0; in DoMirrorPadGrad()
169 MirrorPadMode mode; in Compile()
172 ctx, mode == MirrorPadMode::REFLECT || mode == MirrorPadMode::SYMMETRIC, in Compile()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mirror_pad_mode.cc | 25 MirrorPadMode* value) { in GetNodeAttr()
29 *value = MirrorPadMode::REFLECT; in GetNodeAttr()
31 *value = MirrorPadMode::SYMMETRIC; in GetNodeAttr()
|
| D | mirror_pad_mode.h | 34 enum class MirrorPadMode { enum
48 MirrorPadMode* value);
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | mirror_pad_op.cc | 42 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_operator.cc | 89 static tflite::MirrorPadMode ConvertTFL_MirrorPaddingAttrForOptionWriter( in ConvertTFL_MirrorPaddingAttrForOptionWriter()
91 return llvm::StringSwitch<tflite::MirrorPadMode>(str) in ConvertTFL_MirrorPaddingAttrForOptionWriter()
227 static mlir::Attribute BuildTFL_MirrorPaddingAttr(tflite::MirrorPadMode value, in BuildTFL_MirrorPaddingAttr()
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | enum_mapping.h | 131 inline MirrorPadMode MirrorPaddingModeToSchema(TfLiteMirrorPaddingMode mode) { in MirrorPaddingModeToSchema()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | mirror_pad_test.cc | 27 const TensorData& output, const tflite::MirrorPadMode mode) { in MirrorPadOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_fused_image_transform.cc | 616 MirrorPadMode mode; in FusedResizeConv2DUsingGemmOp()
620 case MirrorPadMode::SYMMETRIC: { in FusedResizeConv2DUsingGemmOp()
624 case MirrorPadMode::REFLECT: { in FusedResizeConv2DUsingGemmOp()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mirror_pad_test.cc | 33 const tflite::MirrorPadMode mode) { in BaseMirrorPadOpModel()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1503 *builder, op.mode == MirrorPadMode::kReflect in WriteOptions()
1504 ? ::tflite::MirrorPadMode::MirrorPadMode_REFLECT in WriteOptions()
1505 : ::tflite::MirrorPadMode::MirrorPadMode_SYMMETRIC); in WriteOptions()
1509 op->mode = options.mode() == ::tflite::MirrorPadMode::MirrorPadMode_REFLECT in ReadOptions()
1510 ? MirrorPadMode::kReflect in ReadOptions()
1511 : MirrorPadMode::kSymmetric; in ReadOptions()
|
| D | operator_test.cc | 748 op.mode = MirrorPadMode::kReflect; in TEST_F()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 2084 enum class MirrorPadMode { kNone, kSymmetric, kReflect };
2098 MirrorPadMode mode;
|
| D | import_tensorflow.cc | 1432 op->mode = toco::MirrorPadMode::kReflect; in ConvertMirrorPadOperator()
1434 op->mode = toco::MirrorPadMode::kSymmetric; in ConvertMirrorPadOperator()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v3a.fbs | 910 enum MirrorPadMode : byte {
918 mode:MirrorPadMode;
|
| D | schema.fbs | 938 enum MirrorPadMode : byte {
946 mode:MirrorPadMode;
|
| D | schema_generated.h | 2977 enum MirrorPadMode {
2984 inline const MirrorPadMode (&EnumValuesMirrorPadMode())[2] {
2985 static const MirrorPadMode values[] = {
3001 inline const char *EnumNameMirrorPadMode(MirrorPadMode e) {
8720 tflite::MirrorPadMode mode;
8731 tflite::MirrorPadMode mode() const {
8732 return static_cast<tflite::MirrorPadMode>(GetField<int8_t>(VT_MODE, 0));
8747 void add_mode(tflite::MirrorPadMode mode) {
8764 tflite::MirrorPadMode mode = tflite::MirrorPadMode_REFLECT) {
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | test_schema.fbs | 901 enum MirrorPadMode : byte {
909 mode:MirrorPadMode;
|