Home
last modified time | relevance | path

Searched refs:UpdateEdge (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/c/
Dpython_api.h43 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst,
Dpython_api.cc58 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, in UpdateEdge() function
Dc_api.cc2523 status->status = graph->graph.UpdateEdge(&new_src.oper->node, new_src.index, in TF_UpdateEdge()
Dc_api_test.cc637 TEST(CAPI, UpdateEdge) { in TEST() argument
/external/tensorflow/tensorflow/core/graph/
Dgraph_test.cc541 TEST_F(GraphTest, UpdateEdge) { in TEST_F() argument
561 TF_EXPECT_OK(graph_.UpdateEdge(a, 0, b, 0)); in TEST_F()
566 TF_EXPECT_OK(graph_.UpdateEdge(a, 0, d, 0)); in TEST_F()
570 Status s = graph_.UpdateEdge(a, 1, d, 0); in TEST_F()
577 s = graph_.UpdateEdge(c, 0, a, 0); in TEST_F()
Dgraph.h572 Status UpdateEdge(Node* new_src, int new_src_index, Node* dst, int dst_index);
Dgraph.cc617 Status Graph::UpdateEdge(Node* new_src, int new_src_index, Node* dst, in UpdateEdge() function in tensorflow::Graph
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils.cc198 TF_RETURN_IF_ERROR(graph->UpdateEdge(identity_node, 0, node, input_idx)); in AddInputIdentity()
260 TF_RETURN_IF_ERROR(graph->UpdateEdge(identity_node, 0, dst, dst_input)); in AddOutputIdentities()
Dlower_while_op.cc463 graph_->UpdateEdge(next_iterations_nodes_[i], 0, merge_nodes_[i], 1)); in UpdateMergeNodes()
/external/tensorflow/tensorflow/compiler/jit/
Ddeadness_analysis_test.cc152 ->UpdateEdge(next_iteration.node(), 0, iv.output.node(), 1) in CreateInductionVariable()
214 ->UpdateEdge(next_iteration.node(), 0, iv.output.node(), 1) in CreateDependentLoopInvariantValue()
677 TF_ASSERT_OK(root.graph()->UpdateEdge( in TEST()
890 TF_ASSERT_OK(root.graph()->UpdateEdge( in TEST()
917 TF_ASSERT_OK(root.graph()->UpdateEdge(div1.induction_var.node(), 0, in TEST()
919 TF_ASSERT_OK(root.graph()->UpdateEdge(div0.induction_var.node(), 0, in TEST()
982 ->UpdateEdge(next_iterations[i].node(), 0, iv.output.node(), 1) in TEST()
Dintroduce_floating_point_jitter_pass.cc111 g->UpdateEdge(jittered_output.node(), 0, e->dst(), e->dst_input())); in IntroduceJitterToTensor()
Dincrease_dynamism_for_auto_jit_pass_test.cc228 TF_ASSERT_OK(root.graph()->UpdateEdge(/*new_src=*/size.node(), in TEST()
318 TF_ASSERT_OK(root.graph()->UpdateEdge(size.node(), 0, slice.node(), 2)); in TEST()
Dxla_cluster_util_test.cc264 return s.graph()->UpdateEdge(next_iteration.node(), 0, merge.output.node(), in MakeLoop()
Dbuild_xla_ops_pass.cc460 TF_RETURN_IF_ERROR(root.graph()->UpdateEdge(identity_n[i].node(), i, n, in PredicateInt32Inputs()
Dresource_operation_safety_analysis_test.cc499 root.graph()->UpdateEdge(next_iteration.node(), 0, iv.output.node(), 1)); in TEST()
/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt229 tensorflow::UpdateEdge
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.h174 Status UpdateEdge(TransposeContext* context, absl::string_view name_format,
Dgeneric_layout_optimizer_transposer.cc385 UpdateEdge(context, in UpdateFaninEdgesWithOp()
428 TF_RETURN_IF_ERROR(UpdateEdge( in UpdateFanoutEdgesWithOp()
486 Status Transposer::UpdateEdge( in UpdateEdge() function in tensorflow::grappler::Transposer
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc554 graph->UpdateEdge(engine_node, conn.port_number, output_node, port)); in CreateTRTNode()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc757 tensorflow::UpdateEdge(graph, new_src, dst, status.get()); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/framework/
Dops.py2273 pywrap_tf_session.UpdateEdge(
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc641 TF_RETURN_IF_ERROR(graph_->UpdateEdge(placeholder_node, 0, edge->dst(), in CreatePlaceholderNodeForFeed()