Home
last modified time | relevance | path

Searched refs:node_outputs (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dinsert_logging.cc72 std::unordered_map<string, std::set<int>> node_outputs; in InsertLogging() local
86 node_outputs[name].insert(output_index); in InsertLogging()
96 if (node_outputs[node.name()].empty()) { in InsertLogging()
130 for (int output_index : node_outputs[node.name()]) { in InsertLogging()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dadd_quant_adjustments.cc44 auto node_outputs = graph->FindOutputs(node->id); in ApplyToNode() local
45 for (auto output_value : node_outputs) { in ApplyToNode()
/external/tensorflow/tensorflow/lite/
Darena_planner.cc163 TfLiteIntArray* node_outputs = node.outputs; in PlanAllocations() local
164 for (int j = 0; j < node_outputs->size; ++j) { in PlanAllocations()
165 int tensor_index = node_outputs->data[j]; in PlanAllocations()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/
Dfuse_auto_input.cc56 auto node_outputs = graph->FindOutputs(node->id); in ApplyToNode() local