Home
last modified time | relevance | path

Searched defs:output_node_names (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/tools/
Dstrip_unused_lib.py32 def strip_unused(input_graph_def, input_node_names, output_node_names, argument
93 output_binary, input_node_names, output_node_names, argument
Doptimize_for_inference_lib.py93 def optimize_for_inference(input_graph_def, input_node_names, output_node_names, argument
419 def fuse_resize_and_conv(input_graph_def, output_node_names): argument
Dfreeze_graph.py80 output_node_names, argument
290 output_node_names, argument
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer_test.cc307 const std::vector<string> output_node_names = {NAME_A_PLUS_B}; in TEST_F() local
318 const std::vector<string> output_node_names = {"softmax"}; in TEST_F() local
344 const std::vector<string> output_node_names = {"softmax"}; in TEST_F() local
379 std::vector<string> output_node_names = {}; in TEST() local
442 std::vector<string> output_node_names = {}; in TEST() local
Dhexagon_graph_execution_test.cc306 std::vector<string> output_node_names; in RunFusedGraph() local
413 std::vector<string> output_node_names = {"softmax"}; in TEST() local
446 std::vector<string> output_node_names = {"softmax"}; in TEST() local
480 std::vector<string> output_node_names = {"softmax"}; in TEST() local
549 std::vector<string> output_node_names = {"softmax"}; in TEST() local
Dgraph_transferer.cc95 const std::vector<string>& output_node_names, in LoadGraphFromProto()
202 const std::vector<string>& output_node_names, const bool is_text_proto, in LoadGraphFromProtoFile()
240 void GraphTransferer::SortParams(const std::vector<string>& output_node_names) { in SortParams()
425 const std::vector<string>& output_node_names) { in RegisterNode()
843 const std::vector<string>& output_node_names) { in RegisterNodeIfAllInputsAreCached()
/external/tensorflow/tensorflow/python/framework/
Dgraph_util_impl.py250 output_node_names, argument
Dconvert_to_constants.py854 output_node_names, argument
1125 output_node_names, argument
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc204 const std::vector<string>& output_node_names, const bool initialize_by_zero, in DryRunInference()
275 std::vector<string> output_node_names; in DryRunInferenceForAllNode() local
338 const std::vector<string>& output_node_names, in ConvertToTensorShapeMap()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc2331 absl::flat_hash_set<absl::string_view> output_node_names; in InferMainFunctionType() local