Searched refs:first_node_id (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | model_xnnpack_test.cc | 49 int first_node_id = interpreter->execution_plan()[0]; in TEST() local
51 interpreter->node_and_registration(first_node_id)->second; in TEST()
77 int first_node_id = interpreter->execution_plan()[0]; in TEST() local
79 interpreter->node_and_registration(first_node_id)->second; in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | special_selector.cc | 33 NodeId first_node_id, in TryDepthwiseConvPlus1x1Conv() argument
36 auto* dw_node = graph.GetNode(first_node_id); in TryDepthwiseConvPlus1x1Conv()
97 const GraphFloat32& graph, NodeId first_node_id, in TryFCFCAdd() argument
100 auto* fc0_node = graph.GetNode(first_node_id); in TryFCFCAdd()
187 const GraphFloat32& graph, NodeId first_node_id, in GPUSubgraphFromGraph() argument
192 TryDepthwiseConvPlus1x1Conv(precision, graph, first_node_id, in GPUSubgraphFromGraph()
200 TryFCFCAdd(gpu_info, precision, graph, first_node_id, tensor_descriptors, in GPUSubgraphFromGraph()
|
| D | special_selector.h | 35 const GraphFloat32& graph, NodeId first_node_id,
|