Searched refs:IsInputArray (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/ |
| H A D | dump_graphviz.cc | 157 if (IsInputArray(model, array_name)) { in GetArrayColorAndShape()
190 if (IsInputArray(model, array_name)) { in GetArrayCompassPt()
549 if (!IsInputArray(model, input) && in DumpOperatorEdges()
|
| H A D | tooling_util.h | 60 bool IsInputArray(const Model& model, const std::string& array_name);
|
| H A D | tooling_util.cc | 101 bool IsInputArray(const Model& model, const std::string& array_name) { in IsInputArray() function
1805 if (IsInputArray(model, array_name) || IsOutputArray(model, array_name)) { in IsAllocatableTransientArray()
2233 if (IsInputArray(model, array_name) || IsOutputArray(model, array_name)) { in IsDiscardableArray()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | remove_unused_op.cc | 41 if (IsInputArray(*model, output)) { in Run()
|
| H A D | quantize.cc | 537 if (IsInputArray(*model, input) && in Run()
606 if (IsInputArray(*model, dequantize_op->inputs[0])) { in Run()
|
| H A D | dequantize.cc | 133 if (IsInputArray(*model, array_name)) { in DequantizeArray()
|