Home
last modified time | relevance | path

Searched refs:op_signature (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/logging/
Dconversion_log_util.cc94 string op_signature; in GetOperatorSignature() local
98 op_signature.append("INPUT:"); in GetOperatorSignature()
102 op_signature.append(ShapeToStringNoSpace(array.shape())); in GetOperatorSignature()
104 op_signature.append("None"); in GetOperatorSignature()
106 op_signature.append(delimiter); in GetOperatorSignature()
107 op_signature.append(ArrayDataTypeName(array.data_type) + delimiter); in GetOperatorSignature()
110 op_signature.append("OUTPUT:"); in GetOperatorSignature()
114 op_signature.append(ShapeToStringNoSpace(array.shape())); in GetOperatorSignature()
116 op_signature.append("None"); in GetOperatorSignature()
118 op_signature.append(delimiter); in GetOperatorSignature()
[all …]
Dgen_html.py41 def get_input_type_from_signature(op_signature): argument
63 start = op_signature.find(":")
64 end = op_signature.find("::OUTPUT")
65 inputs = op_signature[start + 1:end]
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc522 int GetVersion(const OperatorSignature& op_signature) const override { in GetVersion()
524 static_cast<const TocoOperator&>(*op_signature.op); in GetVersion()
684 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST() local
685 const auto key = details::OperatorKey(op_signature, ops_by_type, false); in TEST()
706 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST() local
707 const auto key = details::OperatorKey(op_signature, ops_by_type, false); in TEST()
720 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST() local
721 const auto key = details::OperatorKey(op_signature, ops_by_type, false); in TEST()
735 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST() local
736 const auto key = details::OperatorKey(op_signature, ops_by_type, false); in TEST()
[all …]
Doperator.cc64 const ::tflite::BuiltinOperator op, const OperatorSignature& op_signature) { in GetVersioningOpSig() argument
66 for (auto input_name : op_signature.op->inputs) { in GetVersioningOpSig()
68 if (op_signature.model->HasArray(input_name)) { in GetVersioningOpSig()
69 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersioningOpSig()
74 for (auto output_name : op_signature.op->outputs) { in GetVersioningOpSig()
76 if (op_signature.model->HasArray(output_name)) { in GetVersioningOpSig()
77 const Array& output_array = op_signature.model->GetArray(output_name); in GetVersioningOpSig()
175 int GetVersion(const OperatorSignature& op_signature) const override { in GetVersion()
177 static_cast<const DepthwiseConvOperator&>(*op_signature.op); in GetVersion()
179 GetVersioningOpSig(builtin_op(), op_signature); in GetVersion()
[all …]
Dsimple_operator.h50 int GetVersion(const OperatorSignature& op_signature) const override { in GetVersion() argument
52 GetVersioningOpSig(builtin_op_, op_signature)); in GetVersion()
Dexport.cc113 const ::toco::OperatorSignature& op_signature, in OperatorKey()
117 const ::toco::Operator& op = *op_signature.op; in OperatorKey()
123 version_ = ops_by_type.at(op.type)->GetVersion(op_signature); in OperatorKey()
198 const toco::OperatorSignature op_signature = {op.get(), &model}; in LoadOperatorsMap() local
199 keys.insert(OperatorKey(op_signature, ops_by_type, enable_select_tf_ops)); in LoadOperatorsMap()
312 const toco::OperatorSignature op_signature = {op.get(), &model}; in ExportOperatorCodes() local
314 op_signature, ops_by_type, params.enable_select_tf_ops); in ExportOperatorCodes()
361 const toco::OperatorSignature op_signature = {op.get(), &model}; in ExportOperators() local
362 const auto key = details::OperatorKey(op_signature, ops_by_type, in ExportOperators()
Dbuiltin_operator.h70 int GetVersion(const OperatorSignature& op_signature) const override { in GetVersion() argument
72 GetVersioningOpSig(builtin_op_, op_signature)); in GetVersion()
Dop_version.cc241 OperatorSignature op_signature; in GetMinimumRuntimeVersionForModel() local
242 op_signature.model = &model; in GetMinimumRuntimeVersionForModel()
246 op_signature.op = op.get(); in GetMinimumRuntimeVersionForModel()
247 const int version = op_types_map.at(op->type)->GetVersion(op_signature); in GetMinimumRuntimeVersionForModel()
Doperator.h100 virtual int GetVersion(const OperatorSignature& op_signature) const = 0;
121 const OperatorSignature& op_signature);
Dexport.h96 const ::toco::OperatorSignature& op_signature,