Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dspecial_selector.cc35 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in TryDepthwiseConvPlus1x1Conv() argument
57 if (consumed_nodes->find(conv_node->id) != consumed_nodes->end()) { in TryDepthwiseConvPlus1x1Conv()
89 consumed_nodes->insert(dw_node->id); in TryDepthwiseConvPlus1x1Conv()
90 consumed_nodes->insert(conv_node->id); in TryDepthwiseConvPlus1x1Conv()
99 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in TryFCFCAdd() argument
121 if (consumed_nodes->find(add_node->id) != consumed_nodes->end()) { in TryFCFCAdd()
140 if (consumed_nodes->find(fc1_node->id) != consumed_nodes->end()) { in TryFCFCAdd()
178 consumed_nodes->insert(fc0_node->id); in TryFCFCAdd()
179 consumed_nodes->insert(fc1_node->id); in TryFCFCAdd()
180 consumed_nodes->insert(add_node->id); in TryFCFCAdd()
[all …]
Dspecial_selector.h37 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph,
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc299 std::set<NodeId> consumed_nodes; in ConvertOperations() local
309 if (consumed_nodes.find(node.id) != consumed_nodes.end()) { in ConvertOperations()
326 tensor_descriptors, &consumed_nodes, &gpu_subgraph, in ConvertOperations()
354 consumed_nodes.insert(node.id); in ConvertOperations()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc210 std::set<NodeId> consumed_nodes; in Compile() local
234 tensor_descriptors, &consumed_nodes, &gpu_subgraph, in Compile()
262 consumed_nodes.insert(node.id); in Compile()