Searched refs:enable_select_tf_ops (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.h | 34 bool enable_select_tf_ops = false);
39 bool enable_select_tf_ops = false);
125 bool ShouldExportAsFlexOp(bool enable_select_tf_ops,
|
| D | export.cc | 116 bool enable_select_tf_ops) { in OperatorKey() argument
144 if (ShouldExportAsFlexOp(enable_select_tf_ops, in OperatorKey()
154 } else if (enable_select_tf_ops && !op.tensorflow_node_def.empty()) { in OperatorKey()
195 bool enable_select_tf_ops) { in LoadOperatorsMap() argument
200 keys.insert(OperatorKey(op_signature, ops_by_type, enable_select_tf_ops)); in LoadOperatorsMap()
315 op_signature, ops_by_type, params.enable_select_tf_ops); in ExportOperatorCodes()
364 params.enable_select_tf_ops); in ExportOperators()
434 const auto ops_by_type = BuildOperatorByTypeMap(params.enable_select_tf_ops); in Export()
494 params.enable_select_tf_ops); in Export()
564 if (params.enable_select_tf_ops) { in Export()
|
| D | export.h | 32 bool enable_select_tf_ops = false; member
98 bool enable_select_tf_ops);
173 bool enable_select_tf_ops);
|
| D | export_test.cc | 247 params.enable_select_tf_ops = false; in TEST_F()
261 params.enable_select_tf_ops = false; in TEST_F()
325 params.enable_select_tf_ops = false; in TEST_F()
355 params.enable_select_tf_ops = true; in TEST_F()
434 params_.enable_select_tf_ops = false; in SetAllowedOpSets()
443 params_.enable_select_tf_ops = true; in SetAllowedOpSets()
|
| D | operator.cc | 1604 bool enable_select_tf_ops) in TensorFlowUnsupported() argument
1605 : BaseOperator(name, type), enable_select_tf_ops_(enable_select_tf_ops) {} in TensorFlowUnsupported()
1833 bool enable_select_tf_ops = false) { in BuildOperatorList() argument
1987 enable_select_tf_ops)); in BuildOperatorList()
2090 bool enable_select_tf_ops) { in BuildOperatorByTypeMap() argument
2094 BuildOperatorList(enable_select_tf_ops); in BuildOperatorByTypeMap()
2103 bool enable_select_tf_ops) { in BuildOperatorByNameMap() argument
2107 BuildOperatorList(enable_select_tf_ops); in BuildOperatorByNameMap()
2115 bool ShouldExportAsFlexOp(bool enable_select_tf_ops, in ShouldExportAsFlexOp() argument
2118 if (!enable_select_tf_ops) { in ShouldExportAsFlexOp()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | toco_cmdline_flags.cc | 187 Flag("enable_select_tf_ops", parsed_flags.enable_select_tf_ops.bind(), in ParseTocoFlagsFromCommandLineFlags()
188 parsed_flags.enable_select_tf_ops.default_value(), ""), in ParseTocoFlagsFromCommandLineFlags()
287 READ_TOCO_FLAG(enable_select_tf_ops, FlagRequirement::kNone); in ReadTocoFlagsFromCommandLineFlags()
291 !parsed_toco_flags.enable_select_tf_ops.value()) { in ReadTocoFlagsFromCommandLineFlags()
|
| D | toco_flags.proto | 198 // `enable_select_tf_ops` should always be used with `allow_custom_ops`.
200 optional bool enable_select_tf_ops = 27 [default = false]; field
205 // This will force `enable_select_tf_ops` to true.
206 // `force_select_tf_ops` should always be used with `enable_select_tf_ops`.
|
| D | toco_tooling.cc | 485 params.enable_select_tf_ops = in Export()
486 toco_flags.force_select_tf_ops() || toco_flags.enable_select_tf_ops(); in Export()
|
| D | args.h | 190 Arg<bool> enable_select_tf_ops = Arg<bool>(false); member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
| D | tf_tfl_flatbuffer_helpers.cc | 296 bool emit_select_tf_ops = toco_flags.enable_select_tf_ops(); in ConvertMLIRToTFLiteFlatBuffer()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 356 toco.enable_select_tf_ops = True
|