Home
last modified time | relevance | path

Searched refs:output_node_count (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/
Dhexagon_controller.c168 const int output_node_count = GetOutputNodeCount(); in hexagon_controller_ExecuteGraphWithBuffer() local
169 if (output_node_count <= 0) { in hexagon_controller_ExecuteGraphWithBuffer()
173 hexagon_nn_tensordef outputs[output_node_count]; in hexagon_controller_ExecuteGraphWithBuffer()
175 hexagon_controller_SetAllOutputTensorDef(output_node_count, outputs); in hexagon_controller_ExecuteGraphWithBuffer()
177 nn_id, input_node_count, inputs, output_node_count, outputs); in hexagon_controller_ExecuteGraphWithBuffer()
178 for (int i = 0; i < output_node_count; ++i) { in hexagon_controller_ExecuteGraphWithBuffer()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test.cc104 const int output_node_count, in BuildRemoteFusedGraphExecuteOp() argument
Dremote_fused_graph_execute_utils.cc345 const int output_node_count = output_node_names.size(); in ConvertToTensorShapeMap() local
346 CHECK_EQ(output_node_count, output_tensors.size()); in ConvertToTensorShapeMap()
347 for (int i = 0; i < output_node_count; ++i) { in ConvertToTensorShapeMap()