| /external/tensorflow/tensorflow/core/graph/ |
| D | algorithm_test.cc | 167 g.AddEdge(g.FindNodeId(n3->id()), 0, g.FindNodeId(n1->id()), 1); in TEST()
175 g.sink_node(), g.FindNodeId(n3->id()), g.FindNodeId(n2->id()), in TEST()
176 g.FindNodeId(n1->id()), g.FindNodeId(n0->id()), g.source_node()}; in TEST()
|
| D | graph.h | 587 Node* FindNodeId(int id) const { return nodes_[id]; } in FindNodeId() function
604 Node* source_node() const { return FindNodeId(kSourceId); } in source_node()
605 Node* sink_node() const { return FindNodeId(kSinkId); } in sink_node()
817 if (id_ >= graph_->num_node_ids() || graph_->FindNodeId(id_) != nullptr) {
823 inline Node* NodeIter::operator*() { return graph_->FindNodeId(id_); }
825 inline Node* NodeIter::operator->() { return graph_->FindNodeId(id_); }
|
| D | validate.cc | 70 const Node* n = graph.FindNodeId(i); in ValidateGraphHasNoCycle()
107 nodes_in_cycle.push_back(graph.FindNodeId(i)->name()); in ValidateGraphHasNoCycle()
|
| D | graph_constructor_test.cc | 900 const string& source = graph_.FindNodeId(Graph::kSourceId)->name(); in TEST_F()
901 const string& sink = graph_.FindNodeId(Graph::kSinkId)->name(); in TEST_F()
2437 const string& source = graph_.FindNodeId(Graph::kSourceId)->name(); in TEST_F()
2438 const string& sink = graph_.FindNodeId(Graph::kSinkId)->name(); in TEST_F()
|
| D | graph.cc | 644 const Node* node = FindNodeId(id); in ToGraphDefSubRange()
|
| D | graph_test.cc | 316 Node* node = graph_.FindNodeId(id); in TEST_F()
|
| D | graph_partition.cc | 624 Node* node = g->FindNodeId(i); in AddControlFlow()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_fusion_optimizer.cc | 252 Node* node_from = graph.FindNodeId(from); in Optimize()
266 Node* node_to = graph.FindNodeId(to); in Optimize()
|
| D | mark_for_compilation_pass.cc | 1150 Node* node_from = graph->FindNodeId(from); in RunImpl()
1173 Node* node_to = graph->FindNodeId(to); in RunImpl()
1229 Node* in = graph->FindNodeId(in_id); in RunImpl()
|
| D | resource_operation_safety_analysis.cc | 312 << NodeToString(*g.FindNodeId(incoming_op.first), in ComputeIncompatibleResourceOperationPairs()
|
| D | xla_cluster_util.cc | 58 auto* node = graph.FindNodeId(node_id); in DescribeCycle()
|
| D | encapsulate_util.cc | 137 Node* dst = g->FindNodeId(edges[i].dst_node_id); in PreprocessDataEdgesBetweenOutsideCompilations()
|
| D | encapsulate_xla_computations_pass_test.cc | 165 return graph->FindNodeId(node->id()); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_while.cc | 73 Node* n = g->FindNodeId(i); in Run()
|
| D | graph_execution_state.h | 154 return graph_->FindNodeId(iter->second); in get_node_by_name()
|
| D | constant_folding_test.cc | 184 Node* n1 = g1.FindNodeId(i); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 122 SimpleNode* FindNodeId(int node_id) { in FindNodeId() function in tensorflow::tensorrt::segment::SimpleGraph
156 const auto n = g->FindNodeId(i); in SimpleGraph()
427 SimpleNode* node = graph->FindNodeId(i); in SegmentGraph()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_stream_util_test.cc | 132 Node* n = g.FindNodeId(it.first); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 425 Node* input_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode()
475 Node* input_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode()
608 Node* output_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode()
|
| D | convert_nodes.cc | 4179 auto outside_node = graph->FindNodeId(connection.outside_id); in ConvertSegmentToGraphDef()
4190 graph->FindNodeId(connection.outside_id), in ConvertSegmentToGraphDef()
4195 graph->FindNodeId(connection.outside_id), in ConvertSegmentToGraphDef()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 405 map_defun_node_ = outer_scope_->FindNodeId(node_id); in Initialize()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 789 Node* const node = graph.FindNodeId(id); in ExtendSessionGraphHelper()
1978 Node* node = graph->graph.FindNodeId(*pos); in TF_GraphNextOperation()
2128 auto* node = graph->graph.FindNodeId(i); in GraphImportGraphDefLocked()
2451 Node* new_node = parent->graph.FindNodeId(i); in TF_FinishWhileHelper()
2542 Node* n = g->graph.FindNodeId(i); in TF_AddGradientsWithPrefix()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 1339 Node* s = graph_->FindNodeId(s_id); in DeleteReachableAndDeadNodes()
1376 Node* d = graph_->FindNodeId(d_id); in DeleteReachableAndDeadNodes()
|
| D | tf2xla_util.cc | 148 Node* usage_node = g->FindNodeId(usages[i].dst_node_id); in ReplaceArgUsageWithConstNode()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 492 const Node* node = graph.FindNodeId(i); in BuildTensorShapeMapFromGraph()
|