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<string, std::unique_ptr<BaseOperator>> BuildOperatorByNameMap(
38 std::map<OperatorType, std::unique_ptr<BaseOperator>> BuildOperatorByTypeMap(
71 class BaseOperator {
74 BaseOperator(const 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 | builtin_operator.h | 33 class BuiltinOperator : public BaseOperator {
39 : BaseOperator(::tflite::EnumNameBuiltinOperator(op), type), in BuiltinOperator()
|
| 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()
548 const BaseOperator* base_op = operator_by_type_map.at(op.type).get(); in TEST_F()
781 const BaseOperator* base_op = operator_by_type_map.at(op.type).get(); in SimpleVersioningTest()
803 const BaseOperator* base_op = operator_by_type_map.at(op.type).get(); in SimpleOutputVersioningTest()
858 const BaseOperator* base_op = operator_by_type_map.at(op.type).get(); in TEST_F()
[all …]
|
| D | export.cc | 114 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in OperatorKey()
193 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in LoadOperatorsMap()
294 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in ExportOperatorCodes()
337 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type, in ExportOperators()
343 auto is_tflite_builtin = [](const BaseOperator* op) { in ExportOperators()
367 BaseOperator* tflite_op = tflite_op_it == ops_by_type.end() in ExportOperators()
476 const std::map<OperatorType, std::unique_ptr<BaseOperator>>& ops_by_type) { in Export()
|
| D | operator.cc | 1521 class TensorFlowUnsupported : public BaseOperator {
1525 : BaseOperator(name, type), enable_select_tf_ops_(enable_select_tf_ops) {} in TensorFlowUnsupported()
1755 std::vector<std::unique_ptr<BaseOperator>> BuildOperatorList( in BuildOperatorList()
1757 std::vector<std::unique_ptr<BaseOperator>> ops; in BuildOperatorList()
2010 std::map<OperatorType, std::unique_ptr<BaseOperator>> BuildOperatorByTypeMap( in BuildOperatorByTypeMap()
2012 std::map<OperatorType, std::unique_ptr<BaseOperator>> result; in BuildOperatorByTypeMap()
2014 std::vector<std::unique_ptr<BaseOperator>> ops = in BuildOperatorByTypeMap()
2023 std::map<string, std::unique_ptr<BaseOperator>> BuildOperatorByNameMap( in BuildOperatorByNameMap()
2025 std::map<string, std::unique_ptr<BaseOperator>> result; in BuildOperatorByNameMap()
2027 std::vector<std::unique_ptr<BaseOperator>> ops = in BuildOperatorByNameMap()
|
| D | export_test.cc | 590 std::map<OperatorType, std::unique_ptr<BaseOperator>>
592 std::map<OperatorType, std::unique_ptr<BaseOperator>> result; in BuildFakeOperatorByTypeMap()
594 std::unique_ptr<BaseOperator>(new FakeConvolutionOperator); in BuildFakeOperatorByTypeMap()
|
| D | import.cc | 102 const std::map<string, std::unique_ptr<BaseOperator>>& ops_by_name, in ImportOperators()
|
| /external/tensorflow/tensorflow/lite/toco/logging/ |
| D | conversion_log_util.cc | 90 const std::map<OperatorType, std::unique_ptr<tflite::BaseOperator>>& in GetOperatorSignature()
|