Home
last modified time | relevance | path

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

12

/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/data/
Dserialization_utils.cc59 const string& output_node, Tensor* result) { in FromGraphDef()
278 tstring output_node, serialized_graph_def; in ReadDatasetInternal() local
404 string output_node; in WriteDatasetInternal() local
461 Node* output_node = nullptr; in AsGraphDef() local
Drewrite_utils.cc105 const std::string& output_node = tensor.substr(0, tensor.find(':')); in RemoveFakeSinks() local
176 string output_node; in RewriteDataset() local
Dhash_utils_test.cc164 NodeDef* output_node = g.add_node(); in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_ops.cc149 string output_node; in Compute() local
Dparallel_interleave_dataset_op_test.cc654 Node* output_node = nullptr; in TEST_F() local
695 Node* output_node = nullptr; in TEST_F() local
/external/tensorflow/tensorflow/core/tfrt/utils/
Dgraph_partition_test.cc149 NodeDef* output_node = nullptr; in TEST_F() local
Dgraph_partition.cc242 for (const auto& output_node : output_nodes) { in BuildPartitionedCallOp() local
Dtfrt_graph_execution_state.cc270 for (const auto& output_node : output_nodes) { in PlaceInputOutputNodesOnHost() local
Dtfrt_graph_execution_state_test.cc764 NodeDef* output_node = nullptr; member
/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.cc252 auto* output_node = graph_view.GetNode("Output"); in TEST_F() local
572 auto* output_node = graph_view.GetNode("output"); in TEST_F() local
Dremapper.cc2602 auto* output_node = in AddFusedMatMulBiasAddAndGelu() local
2649 auto* output_node = in AddMklLayerNorm() local
3144 auto* output_node = in AddFusedBatchMatMul() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc223 auto output_node = edge->dst(); in GetEngineInfo() local
548 Node* output_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
/external/tensorflow/tensorflow/core/grappler/
Dutils.h190 auto output_node = nodes_[NodeName(output_name)]; in AddOutput() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc294 Node* output_node = le->dst(); in BuildXlaLaunchOps() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dtf_to_tfrt_data.cc289 string output_node; in TFDataGraphDefToTFDataMLIR() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc135 Node* output_node; in ConstructFunctionGraph() 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.cc97 Node* output_node = output_edge->dst(); in ResourceOnlyUsedForTPUExecuteInLoop() local
/external/XNNPACK/src/
Dsubgraph.c912 struct xnn_node* output_node = subgraph->nodes + subgraph->num_nodes - 1; in xnn_subgraph_rewrite_for_fp16() local
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_test.cc1050 const string output_node = "Func/b/output/_1"; in TEST_F() local
1125 const string output_node = "Func/b/output/_1"; in TEST_F() local
1213 const string output_node = "Func/b/output/_1"; in TEST_F() local
/external/crosvm/devices/src/virtio/
Diommu.rs120 output_node: u16, field
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc854 for (auto* output_node : port_num_output_pair.second) { in GetOutputNodes() local
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc479 void AddInput(int node_index, int output_node, int output_index) { in AddInput()

12