Home
last modified time | relevance | path

Searched refs:output_node (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Drewrite_utils.cc75 const string& output_node = tensor.substr(0, tensor.find(':')); in RemoveFakeSinks() local
76 if (identity_map.find(output_node) != identity_map.end()) { in RemoveFakeSinks()
78 identity_map.at(output_node); in RemoveFakeSinks()
85 GraphDef* graph_def, string* output_node) { in ApplyRewrites() argument
93 node->add_input(*output_node); in ApplyRewrites()
95 *output_node = node->name(); in ApplyRewrites()
113 node_list->add_value(*output_node); in ApplyRewrites()
152 string output_node; in RewriteDataset() local
154 AsGraphDefMinimal(ctx, input, &input_list, &graph_def, &output_node)); in RewriteDataset()
158 ApplyRewrites(ctx, config_factory, &graph_def, &output_node)); in RewriteDataset()
[all …]
Dserialization_utils.cc100 Node* output_node = nullptr; in AsGraphDef() local
102 db.AddInputDataset(&serialization_ctx, dataset, &output_node)); in AsGraphDef()
105 ops::UnaryOp("_Retval", output_node, in AsGraphDef()
Ddataset_ops.cc127 string output_node; in Compute() local
130 output_node = node.input(0); in Compute()
153 graph_runner.Run(&graph, flr, {}, {output_node}, &outputs)); in Compute()
/external/tensorflow/tensorflow/python/tools/
Dstrip_unused_test.py49 output_node = math_ops.multiply(
51 math_ops.add(output_node, 2.0, name="later_node")
53 output = self.evaluate(output_node)
98 output_node = sess.graph.get_tensor_by_name("output_node:0")
99 output = sess.run(output_node, feed_dict={input_node: [10.0]})
112 output_node = math_ops.multiply(
114 math_ops.add(output_node, 2.0, name="later_node")
116 output = self.evaluate(output_node)
156 output_node = sess.graph.get_tensor_by_name("output_node:0")
157 output = sess.run(output_node,
Dfreeze_graph_test.py64 output_node = math_ops.multiply(variable_node, 2.0, name="output_node")
68 output = sess.run(output_node)
118 output_node = sess.graph.get_tensor_by_name("output_node:0")
119 output = sess.run(output_node)
180 output_node = math_ops.multiply(variable_node, 2.0, name="output_node")
184 output = sess.run(output_node)
220 output_node = sess.graph.get_tensor_by_name("output_node:0")
221 output = sess.run(output_node)
274 output_node = sess.graph.get_tensor_by_name("output_node:0")
275 output = sess.run(output_node, feed_dict={input_node: [example]})
/external/autotest/client/site_tests/audio_Aplay/
Daudio_Aplay.py116 output_node = audio_spec.get_headphone_node(utils.get_board())
118 output_node = "INTERNAL_SPEAKER"
119 logging.debug("Test output device %s", output_node)
121 cras_utils.set_single_selected_output_node(output_node)
125 if cras_device_type != output_node:
/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h168 NodeDef output_node = create_node(/*name=*/absl::StrFormat("out%05d", i)); in CreateFaninFanoutNodeGraph() local
171 output_node.add_input(input_node_index); in CreateFaninFanoutNodeGraph()
172 *graph.add_node() = std::move(output_node); in CreateFaninFanoutNodeGraph()
177 NodeDef output_node = in CreateFaninFanoutNodeGraph() local
179 output_node.add_input(controlled_fanout_input); in CreateFaninFanoutNodeGraph()
180 *graph.add_node() = std::move(output_node); in CreateFaninFanoutNodeGraph()
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants.py416 output_node = self.converted_self().node
417 output_node.Clear()
418 output_node.name = self._node.name
419 output_node.op = "GatherV2"
420 output_node.input.extend(
422 output_node.attr["Tparams"].CopyFrom(self._node.attr["dtype"])
423 output_node.attr["Tindices"].CopyFrom(self._node.attr["Tindices"])
424 output_node.attr["Taxis"].CopyFrom(axis_dtype)
426 output_node.attr["_class"].CopyFrom(self._node.attr["_class"])
433 output_node = self.converted_self().node
[all …]
Dconvert_to_constants_test.py571 output_node = math_ops.multiply(variable_node, 2.0, name="output_node")
574 output = self.evaluate(output_node)
593 output_node = sess.graph.get_tensor_by_name("output_node:0")
594 output = self.evaluate(output_node)
606 output_node = array_ops.identity(variable_node, name="output_node")
611 output = self.evaluate(output_node)
639 output_node = sess.graph.get_tensor_by_name("output_node:0")
641 output = self.evaluate(output_node)
643 output = self.evaluate(output_node)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dfeed-control-dep.pbtxt1 …-tf-input-arrays=input -tf-input-data-types=DT_FLOAT -tf-output-arrays=output_node -o - | FileChec…
46 name: "output_node"
66 # CHECK-SAME: outputs = "output_node"
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc218 auto output_node = edge->dst(); in GetEngineInfo() local
219 if (output_node->IsSink() || segment_nodes.count(output_node)) { in GetEngineInfo()
225 info->connections.emplace_back(output_node->name(), output_node->id(), in GetEngineInfo()
242 output_node->name(), output_node->id(), edge->dst_input(), in GetEngineInfo()
537 Node* output_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
539 if (!output_node) { in CreateTRTNode()
541 conn.outside_node_name, &output_node, &port); in CreateTRTNode()
545 << output_node->name(); in CreateTRTNode()
547 graph->AddControlEdge(engine_node, output_node); in CreateTRTNode()
550 << conn.port_number << " to " << output_node->name() << ":" in CreateTRTNode()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Doptimize-arg-operand-constraint.mlir7 attributes {tf.entry_function = {inputs = "input", outputs = "output_node"}} {
11 …%3 = "tf.Mul"(%2, %0) {T = "tfdtype$DT_FLOAT", device = "", name = "output_node"} : (tensor<1xf32>…
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc300 Node* output_node = le->dst(); in BuildXlaLaunchOps() local
302 TF_RET_CHECK(output_node->type_string() == kXlaClusterOutput) in BuildXlaLaunchOps()
304 nodes_to_remove.push_back(output_node); in BuildXlaLaunchOps()
306 for (const Edge* oe : output_node->out_edges()) { in BuildXlaLaunchOps()
311 output_types[le->src_output()] = output_node->input_type(0); in BuildXlaLaunchOps()
313 AddControlOutputs(*output_node, &control_outputs); in BuildXlaLaunchOps()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc872 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
873 ASSERT_NE(output_node, nullptr); in TEST_F()
874 ASSERT_EQ(output_node->NumRegularFanins(), 1); in TEST_F()
875 VerifyRegularFaninMatch(output_node, 0, output_transpose_node->GetName(), 0); in TEST_F()
965 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
966 ASSERT_NE(output_node, nullptr); in TEST_F()
967 ASSERT_EQ(output_node->NumRegularFanins(), 1); in TEST_F()
968 VerifyRegularFaninMatch(output_node, 0, bag_node->GetName(), 0); in TEST_F()
998 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
999 ASSERT_NE(output_node, nullptr); in TEST_F()
[all …]
Dgeneric_layout_optimizer_test.cc269 auto* output_node = graph_view.GetNode("Output"); in TEST_F() local
270 ASSERT_NE(output_node, nullptr); in TEST_F()
271 ASSERT_EQ(output_node->NumRegularFanins(), 1); in TEST_F()
272 VerifyRegularFaninMatch(output_node, 0, output_transpose_node->GetName(), 0); in TEST_F()
602 auto* output_node = graph_view.GetNode("output"); in TEST_F() local
603 ASSERT_NE(output_node, nullptr); in TEST_F()
604 ASSERT_EQ(output_node->NumRegularFanins(), 1); in TEST_F()
605 VerifyRegularFaninMatch(output_node, 0, output_out_transpose_node->GetName(), in TEST_F()
Ddependency_optimizer.cc218 for (NodeDef* output_node : output_nodes) { in BypassingNodeIsBeneficial()
219 num_cross_out += static_cast<int>(output_node->device() != node_dev); in BypassingNodeIsBeneficial()
226 for (NodeDef* output_node : output_nodes) { in BypassingNodeIsBeneficial()
228 static_cast<int>(input_node->device() != output_node->device()); in BypassingNodeIsBeneficial()
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py541 output_node = _copy.deepcopy(self.node)
542 del output_node.input[:]
543 output_node.input.append(_tensorflow_output_name(fused_op_name, index))
544 out_graphdef.node.extend([output_node])
663 output_node = _copy.deepcopy(discrete)
664 del output_node.input[:]
665 output_node.input.append(_tensorflow_output_name(stack_node.name, idx))
666 out_graphdef.node.extend([output_node])
/external/tensorflow/tensorflow/core/grappler/
Dutils.h190 auto output_node = nodes_[NodeName(output_name)]; in AddOutput() local
191 DCHECK(output_node) << "Output node " << output_name in AddOutput()
193 outputs_[node_name].insert(output_node); in AddOutput()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc136 Node* output_node; in ConstructFunctionGraph() local
138 NodeBuilder(output_node_builder).Finalize(&g, &output_node)); in ConstructFunctionGraph()
140 g.AddEdge(function_node, i, output_node, 0); in ConstructFunctionGraph()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc353 Node* output_node = map_defun_fn_->ret_nodes.at(output_position); in VectorizeHelper() local
355 << output_node->DebugString() << "\nError: " << s; in VectorizeHelper()
356 unconvertible_.insert(output_node); in VectorizeHelper()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc96 Node* output_node = output_edge->dst(); in ResourceOnlyUsedForTPUExecuteInLoop() local
97 if (output_edge->IsControlEdge() || output_node->IsExit()) continue; in ResourceOnlyUsedForTPUExecuteInLoop()
102 output_node)) { in ResourceOnlyUsedForTPUExecuteInLoop()
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc242 for (int output_node : output_nodes) { in GetStopBackpropNodes() local
243 if (internal_outputs.find(output_node) == internal_outputs.end()) { in GetStopBackpropNodes()
244 stop_backprop_nodes.insert(output_node); in GetStopBackpropNodes()
/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
2315 output_node = new_output(aio, control, section->name); in alsa_iodev_ucm_add_nodes_and_jacks()
2316 if (!output_node) in alsa_iodev_ucm_add_nodes_and_jacks()
2318 output_node->pcm_name = strdup(section->pcm_name); in alsa_iodev_ucm_add_nodes_and_jacks()
2334 if (output_node) { in alsa_iodev_ucm_add_nodes_and_jacks()
2335 output_node->jack = jack; in alsa_iodev_ucm_add_nodes_and_jacks()
2336 if (!output_node->volume_curve) in alsa_iodev_ucm_add_nodes_and_jacks()
2337 output_node->volume_curve = in alsa_iodev_ucm_add_nodes_and_jacks()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc849 for (auto* output_node : port_num_output_pair.second) { in GetOutputNodes() local
850 auto& output_state = node_map_[output_node]; in GetOutputNodes()
857 IsMerge(*output_node)) { in GetOutputNodes()
860 output_nodes->push_back(output_node); in GetOutputNodes()
861 VLOG(3) << " Add output: " << output_node->name(); in GetOutputNodes()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_test.cc1057 const string output_node = "Func/b/output/_1"; in TEST_F() local
1071 NDef(output_node, "Identity", {"b/ret"}, {{"T", DT_FLOAT}}), in TEST_F()
1074 NDef("ret", "_Retval", {output_node, "^c"}, in TEST_F()
1094 NDef(output_node, "Identity", {"b/ret"}, {{"T", DT_FLOAT}}), in TEST_F()
1097 NDef("ret", "_Retval", {output_node, "^c"}, in TEST_F()
1132 const string output_node = "Func/b/output/_1"; in TEST_F() local
1143 NDef(output_node, "Identity", {"b/ret"}, {{"T", DT_FLOAT}}), in TEST_F()
1164 {output_node, "^" + output_control_node}, in TEST_F()
1220 const string output_node = "Func/b/output/_1"; in TEST_F() local
1231 NDef(output_node, "Identity", {"b/ret"}, {{"T", DT_FLOAT}}, in TEST_F()
[all …]

12