Searched refs:enable_select_tf_ops (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | export.cc | 111 bool enable_select_tf_ops) { in OperatorKey() argument
139 if (ShouldExportAsFlexOp(enable_select_tf_ops, in OperatorKey()
149 } else if (enable_select_tf_ops && !op.tensorflow_node_def.empty()) { in OperatorKey()
190 bool enable_select_tf_ops) { in LoadOperatorsMap() argument
195 keys.insert(OperatorKey(op_signature, ops_by_type, enable_select_tf_ops)); in LoadOperatorsMap()
309 op_signature, ops_by_type, params.enable_select_tf_ops); in ExportOperatorCodes()
358 params.enable_select_tf_ops); in ExportOperators()
430 const auto ops_by_type = BuildOperatorByTypeMap(params.enable_select_tf_ops); in Export()
445 params.enable_select_tf_ops); in Export()
511 if (params.enable_select_tf_ops) { in Export()
|
| D | operator.h | 33 bool enable_select_tf_ops = false);
38 bool enable_select_tf_ops = false);
118 bool ShouldExportAsFlexOp(bool enable_select_tf_ops,
|
| D | export.h | 29 bool enable_select_tf_ops = false; member
93 bool enable_select_tf_ops);
168 bool enable_select_tf_ops);
|
| D | export_test.cc | 221 params.enable_select_tf_ops = false; in TEST_F()
251 params_.enable_select_tf_ops = false; in SetAllowedOpSets()
260 params_.enable_select_tf_ops = true; in SetAllowedOpSets()
|
| D | operator.cc | 1990 bool enable_select_tf_ops) in TensorFlowUnsupported() argument
1991 : BaseOperator(name, type), enable_select_tf_ops_(enable_select_tf_ops) {} in TensorFlowUnsupported()
2336 bool enable_select_tf_ops = false) { in BuildOperatorList() argument
2487 enable_select_tf_ops)); in BuildOperatorList()
2566 bool enable_select_tf_ops) { in BuildOperatorByTypeMap() argument
2570 BuildOperatorList(enable_select_tf_ops); in BuildOperatorByTypeMap()
2579 bool enable_select_tf_ops) { in BuildOperatorByNameMap() argument
2583 BuildOperatorList(enable_select_tf_ops); in BuildOperatorByNameMap()
2591 bool ShouldExportAsFlexOp(bool enable_select_tf_ops, in ShouldExportAsFlexOp() argument
2594 if (!enable_select_tf_ops) { in ShouldExportAsFlexOp()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | toco_cmdline_flags.cc | 171 Flag("enable_select_tf_ops", parsed_flags.enable_select_tf_ops.bind(), in ParseTocoFlagsFromCommandLineFlags()
172 parsed_flags.enable_select_tf_ops.default_value(), ""), in ParseTocoFlagsFromCommandLineFlags()
270 READ_TOCO_FLAG(enable_select_tf_ops, FlagRequirement::kNone); in ReadTocoFlagsFromCommandLineFlags()
274 !parsed_toco_flags.enable_select_tf_ops.value()) { in ReadTocoFlagsFromCommandLineFlags()
|
| D | toco_flags.proto | 197 // `enable_select_tf_ops` should always be used with `allow_custom_ops`.
199 optional bool enable_select_tf_ops = 27 [default = false]; field
204 // This will force `enable_select_tf_ops` to true.
205 // `force_select_tf_ops` should always be used with `enable_select_tf_ops`.
|
| D | toco_tooling.cc | 449 params.enable_select_tf_ops = in Export()
450 toco_flags.force_select_tf_ops() || toco_flags.enable_select_tf_ops(); in Export()
|
| D | args.h | 186 Arg<bool> enable_select_tf_ops = Arg<bool>(false); member
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 342 toco.enable_select_tf_ops = True
344 toco.enable_select_tf_ops = True
|