Home
last modified time | relevance | path

Searched defs:output_node (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Drewrite_utils.cc75 const string& output_node = tensor.substr(0, tensor.find(':')); in RemoveFakeSinks() local
85 GraphDef* graph_def, string* output_node) { in ApplyRewrites()
152 string output_node; in RewriteDataset() local
Dserialization_utils.cc100 Node* output_node = nullptr; in AsGraphDef() local
Ddataset_ops.cc127 string output_node; in Compute() local
Dhash_utils_test.cc161 NodeDef* output_node = g.add_node(); in TEST_F() local
/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h168 NodeDef output_node = create_node(/*name=*/absl::StrFormat("out%05d", i)); in CreateFaninFanoutNodeGraph() local
177 NodeDef output_node = in CreateFaninFanoutNodeGraph() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc872 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
965 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
998 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1051 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1150 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1336 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1384 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1451 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1499 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1651 auto* output_node = context.graph_view->GetNode("i2"); in TEST_F() local
[all …]
Dgeneric_layout_optimizer_test.cc269 auto* output_node = graph_view.GetNode("Output"); in TEST_F() local
602 auto* output_node = graph_view.GetNode("output"); in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc218 auto output_node = edge->dst(); in GetEngineInfo() local
537 Node* output_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc300 Node* output_node = le->dst(); in BuildXlaLaunchOps() local
/external/tensorflow/tensorflow/core/grappler/
Dutils.h190 auto output_node = nodes_[NodeName(output_name)]; in AddOutput() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc136 Node* output_node; in ConstructFunctionGraph() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc353 Node* output_node = map_defun_fn_->ret_nodes.at(output_position); in VectorizeHelper() local
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc242 for (int output_node : output_nodes) { in GetStopBackpropNodes() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc96 Node* output_node = output_edge->dst(); in ResourceOnlyUsedForTPUExecuteInLoop() local
Ddistributed_tpu_rewrite_pass.cc2133 Node* output_node = partitioned_output_nodes[i]; in AssignArgsAndRetvalsToCores() local
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_test.cc1057 const string output_node = "Func/b/output/_1"; in TEST_F() local
1132 const string output_node = "Func/b/output/_1"; in TEST_F() local
1220 const string output_node = "Func/b/output/_1"; in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc849 for (auto* output_node : port_num_output_pair.second) { in GetOutputNodes() local
Dgraph_properties.cc902 NodeDef* output_node = output_nodes[output_arg.node_name]; in UpdateFunction() local
/external/adhd/cras/src/server/
Dcras_alsa_io.c2286 struct alsa_output_node *output_node = NULL; in alsa_iodev_ucm_add_nodes_and_jacks() local
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc456 void AddInput(int node_index, int output_node, int output_index) { in AddInput()