Searched refs:BaseOperator (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.h | 28 class BaseOperator; variable
33 std::map<std::string, std::unique_ptr<BaseOperator>> BuildOperatorByNameMap(
38 std::map<OperatorType, std::unique_ptr<BaseOperator>> BuildOperatorByTypeMap(
71 class BaseOperator {
74 BaseOperator(const std::string& name, OperatorType type) in BaseOperator() function
76 virtual ~BaseOperator() = default;
|
| D | simple_operator.h | 32 class SimpleOperator : public BaseOperator {
34 using BaseOperator::BaseOperator;
37 : BaseOperator(::tflite::EnumNameBuiltinOperator(op), type), in SimpleOperator()
|
| D | custom_operator.h | 33 class CustomOperator : public BaseOperator {
36 using BaseOperator::BaseOperator;
|
| D | export.h | 45 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type);
64 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type) { in Export()
97 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type,
172 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type,
|
| D | operator_test.cc | 33 const BaseOperator& GetOperator(const std::string& name, OperatorType type) { in GetOperator()
34 using OpsByName = std::map<std::string, std::unique_ptr<BaseOperator>>; in GetOperator()
35 using OpsByType = std::map<OperatorType, std::unique_ptr<BaseOperator>>; in GetOperator()
42 BaseOperator* op1 = by_name->at(name).get(); in GetOperator()
47 BaseOperator* op2 = by_type->at(type).get(); in GetOperator()
59 std::unique_ptr<T> SerializeAndDeserialize(const BaseOperator& op, in SerializeAndDeserialize()
562 const BaseOperator* base_op = operator_by_type_map.at(op.type).get(); in TEST_F()
800 const BaseOperator* base_op = operator_by_type_map.at(op.type).get(); in SimpleVersioningTest()
822 const BaseOperator* base_op = operator_by_type_map.at(op.type).get(); in SimpleOutputVersioningTest()
865 const BaseOperator* base_op = operator_by_type_map.at(op.type).get(); in TEST_F()
[all …]
|
| D | builtin_operator.h | 33 class BuiltinOperator : public BaseOperator {
39 : BaseOperator(::tflite::EnumNameBuiltinOperator(op), type), in BuiltinOperator()
|
| D | export.cc | 115 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in OperatorKey()
194 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in LoadOperatorsMap()
295 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in ExportOperatorCodes()
338 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in ExportOperators()
344 auto is_tflite_builtin = [](const BaseOperator* op) { in ExportOperators()
368 BaseOperator* tflite_op = tflite_op_it == ops_by_type.end() in ExportOperators()
485 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type) { in Export()
|
| D | operator.cc | 1596 class TensorFlowUnsupported : public BaseOperator {
1600 : BaseOperator(name, type), enable_select_tf_ops_(enable_select_tf_ops) {} in TensorFlowUnsupported()
1827 std::vector<std::unique_ptr<BaseOperator>> BuildOperatorList( in BuildOperatorList()
1829 std::vector<std::unique_ptr<BaseOperator>> ops; in BuildOperatorList()
2084 std::map<OperatorType, std::unique_ptr<BaseOperator>> BuildOperatorByTypeMap( in BuildOperatorByTypeMap()
2086 std::map<OperatorType, std::unique_ptr<BaseOperator>> result; in BuildOperatorByTypeMap()
2088 std::vector<std::unique_ptr<BaseOperator>> ops = in BuildOperatorByTypeMap()
2097 std::map<std::string, std::unique_ptr<BaseOperator>> BuildOperatorByNameMap( in BuildOperatorByNameMap()
2099 std::map<std::string, std::unique_ptr<BaseOperator>> result; in BuildOperatorByNameMap()
2101 std::vector<std::unique_ptr<BaseOperator>> ops = in BuildOperatorByNameMap()
|
| D | export_test.cc | 596 std::map<OperatorType, std::unique_ptr<BaseOperator>>
598 std::map<OperatorType, std::unique_ptr<BaseOperator>> result; in BuildFakeOperatorByTypeMap()
600 std::unique_ptr<BaseOperator>(new FakeConvolutionOperator); in BuildFakeOperatorByTypeMap()
|
| D | import.cc | 103 const std::map<std::string, std::unique_ptr<BaseOperator>>& ops_by_name, in ImportOperators()
|
| /external/tensorflow/tensorflow/lite/toco/logging/ |
| D | conversion_log_util.cc | 91 const std::map<OperatorType, std::unique_ptr<tflite::BaseOperator>>& in GetOperatorSignature()
|