Searched refs:BaseOperator (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.h | 27 class BaseOperator; variable
32 std::map<string, std::unique_ptr<BaseOperator>> BuildOperatorByNameMap(
37 std::map<OperatorType, std::unique_ptr<BaseOperator>> BuildOperatorByTypeMap(
70 class BaseOperator {
73 BaseOperator(const string& name, OperatorType type) in BaseOperator() function
75 virtual ~BaseOperator() = default;
|
| D | simple_operator.h | 32 class SimpleOperator : public BaseOperator {
34 using BaseOperator::BaseOperator;
|
| D | custom_operator.h | 33 class CustomOperator : public BaseOperator {
36 using BaseOperator::BaseOperator;
|
| D | export.h | 42 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type);
60 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type) { in Export()
92 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type,
167 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type,
|
| D | builtin_operator.h | 33 class BuiltinOperator : public BaseOperator {
39 : BaseOperator(::tflite::EnumNameBuiltinOperator(op), type) {} in BuiltinOperator()
|
| D | export.cc | 110 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in OperatorKey()
189 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in LoadOperatorsMap()
289 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in ExportOperatorCodes()
332 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in ExportOperators()
338 auto is_tflite_builtin = [](const BaseOperator* op) { in ExportOperators()
362 BaseOperator* tflite_op = tflite_op_it == ops_by_type.end() in ExportOperators()
437 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type) { in Export()
|
| D | operator_test.cc | 34 const BaseOperator& GetOperator(const string& name, OperatorType type) { in GetOperator()
35 using OpsByName = std::map<string, std::unique_ptr<BaseOperator>>; in GetOperator()
36 using OpsByType = std::map<OperatorType, std::unique_ptr<BaseOperator>>; in GetOperator()
43 BaseOperator* op1 = by_name->at(name).get(); in GetOperator()
48 BaseOperator* op2 = by_type->at(type).get(); in GetOperator()
60 std::unique_ptr<T> SerializeAndDeserialize(const BaseOperator& op, in SerializeAndDeserialize()
715 const BaseOperator* base_op = operator_by_type_map.at(op.type).get(); in SimpleVersioningTest()
737 const BaseOperator* base_op = operator_by_type_map.at(op.type).get(); in SimpleOutputVersioningTest()
853 const BaseOperator* op = in TEST_F()
|
| D | export_test.cc | 416 std::map<OperatorType, std::unique_ptr<BaseOperator>>
418 std::map<OperatorType, std::unique_ptr<BaseOperator>> result; in BuildFakeOperatorByTypeMap()
420 std::unique_ptr<BaseOperator>(new FakeConvolutionOperator); in BuildFakeOperatorByTypeMap()
|
| D | operator.cc | 1987 class TensorFlowUnsupported : public BaseOperator {
1991 : BaseOperator(name, type), enable_select_tf_ops_(enable_select_tf_ops) {} in TensorFlowUnsupported()
2335 std::vector<std::unique_ptr<BaseOperator>> BuildOperatorList( in BuildOperatorList()
2337 std::vector<std::unique_ptr<BaseOperator>> ops; in BuildOperatorList()
2565 std::map<OperatorType, std::unique_ptr<BaseOperator>> BuildOperatorByTypeMap( in BuildOperatorByTypeMap()
2567 std::map<OperatorType, std::unique_ptr<BaseOperator>> result; in BuildOperatorByTypeMap()
2569 std::vector<std::unique_ptr<BaseOperator>> ops = in BuildOperatorByTypeMap()
2578 std::map<string, std::unique_ptr<BaseOperator>> BuildOperatorByNameMap( in BuildOperatorByNameMap()
2580 std::map<string, std::unique_ptr<BaseOperator>> result; in BuildOperatorByNameMap()
2582 std::vector<std::unique_ptr<BaseOperator>> ops = in BuildOperatorByNameMap()
|
| D | import.cc | 102 const std::map<string, std::unique_ptr<BaseOperator>>& ops_by_name, in ImportOperators()
|