Searched refs:IsInputArray (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | remove_unused_op.cc | 41 if (IsInputArray(*model, output)) { in Run()
|
| D | dequantize.cc | 133 if (IsInputArray(*model, array_name)) { in DequantizeArray()
|
| D | quantize.cc | 485 if (IsInputArray(*model, input) && in Run()
557 if (IsInputArray(*model, dequantize_op->inputs[0])) { in Run()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | dump_graphviz.cc | 153 if (IsInputArray(model, array_name)) { in GetArrayColorAndShape()
185 if (IsInputArray(model, array_name)) { in GetArrayCompassPt()
542 if (!IsInputArray(model, input) && in DumpOperatorEdges()
|
| D | tooling_util.h | 62 bool IsInputArray(const Model& model, const string& array_name);
|
| D | tooling_util.cc | 99 bool IsInputArray(const Model& model, const string& array_name) { in IsInputArray() function
1779 if (IsInputArray(model, array_name) || IsOutputArray(model, array_name)) { in IsAllocatableTransientArray()
2187 if (IsInputArray(model, array_name) || IsOutputArray(model, array_name)) { in IsDiscardableArray()
|