Searched refs:BaseOperator (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| D | operator.h | 26 class BaseOperator; variable
29 std::map<string, std::unique_ptr<BaseOperator>> BuildOperatorByNameMap();
33 std::map<OperatorType, std::unique_ptr<BaseOperator>> BuildOperatorByTypeMap();
60 class BaseOperator {
63 BaseOperator(const string& name, OperatorType type) in BaseOperator() function
65 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 | builtin_operator.h | 33 class BuiltinOperator : public BaseOperator {
39 : BaseOperator(::tflite::EnumNameBuiltinOperator(op), type) {} in BuiltinOperator()
|
| D | operator.cc | 660 class TensorFlowUnsupported : public BaseOperator {
662 using BaseOperator::BaseOperator;
768 std::vector<std::unique_ptr<BaseOperator>> BuildOperatorList() { in BuildOperatorList()
769 std::vector<std::unique_ptr<BaseOperator>> ops; in BuildOperatorList()
867 std::map<OperatorType, std::unique_ptr<BaseOperator>> BuildOperatorByTypeMap() { in BuildOperatorByTypeMap()
868 std::map<OperatorType, std::unique_ptr<BaseOperator>> result; in BuildOperatorByTypeMap()
870 std::vector<std::unique_ptr<BaseOperator>> ops = BuildOperatorList(); in BuildOperatorByTypeMap()
878 std::map<string, std::unique_ptr<BaseOperator>> BuildOperatorByNameMap() { in BuildOperatorByNameMap()
879 std::map<string, std::unique_ptr<BaseOperator>> result; in BuildOperatorByNameMap()
881 std::vector<std::unique_ptr<BaseOperator>> ops = BuildOperatorList(); in BuildOperatorByNameMap()
|
| D | operator_test.cc | 33 const BaseOperator& GetOperator(const string& name, OperatorType type) { in GetOperator()
34 using OpsByName = std::map<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()
|
| D | export.cc | 170 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in ExportOperatorCodes()
230 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in ExportOperators()
|
| D | import.cc | 97 const std::map<string, std::unique_ptr<BaseOperator>>& ops_by_name, in ImportOperators()
|