Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dpython_api.h48 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst,
Dpython_api.cc65 void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, in UpdateEdge() function
Dc_api.cc2548 status->status = graph->graph.UpdateEdge(&new_src.oper->node, new_src.index, in TF_UpdateEdge()
Dc_api_test.cc640 TEST(CAPI, UpdateEdge) { in TEST() argument
/external/tensorflow/tensorflow/core/common_runtime/
Dforward_type_inference_test.cc147 TF_ASSERT_OK(root.graph()->UpdateEdge(next, 0, merge, 1)); in TEST()
304 TF_ASSERT_OK(graph->UpdateEdge(an_copy, 0, m_copy, 1)); in TEST()
305 TF_ASSERT_OK(graph->UpdateEdge(tn_copy, 0, m_copy, 0)); in TEST()
370 TF_ASSERT_OK(root.graph()->UpdateEdge(next, 0, merge, 1)); in TEST()
Dplacer_inspection_required_ops_utils.cc190 TF_RETURN_IF_ERROR(graph->UpdateEdge(identity_node, 0, node, input_idx)); in AddInputIdentity()
248 TF_RETURN_IF_ERROR(graph->UpdateEdge(identity_node, 0, dst, dst_input)); in AddOutputIdentities()
Dlower_while_op.cc465 graph_->UpdateEdge(next_iterations_nodes_[i], 0, merge_nodes_[i], 1)); in UpdateMergeNodes()
/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.h602 Status UpdateEdge(Node* new_src, int new_src_index, Node* dst, int dst_index);
Dgraph.cc685 Status Graph::UpdateEdge(Node* new_src, int new_src_index, Node* dst, in UpdateEdge() function in tensorflow::Graph
/external/tensorflow/tensorflow/compiler/jit/
Ddeadness_analysis_test.cc152 ->UpdateEdge(next_iteration.node(), 0, iv.output.node(), 1) in CreateInductionVariable()
215 ->UpdateEdge(next_iteration.node(), 0, iv.output.node(), 1) in CreateDependentLoopInvariantValue()
678 TF_ASSERT_OK(root.graph()->UpdateEdge( in TEST()
891 TF_ASSERT_OK(root.graph()->UpdateEdge( in TEST()
918 TF_ASSERT_OK(root.graph()->UpdateEdge(div1.induction_var.node(), 0, in TEST()
920 TF_ASSERT_OK(root.graph()->UpdateEdge(div0.induction_var.node(), 0, in TEST()
983 ->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.cc229 TF_ASSERT_OK(root.graph()->UpdateEdge(/*new_src=*/size.node(), in TEST()
319 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.cc459 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/core/tfrt/utils/
Dgraph_partition.cc124 subgraph->UpdateEdge(arg_node, 0, edge->dst(), edge->dst_input())); in PrepareSubgraphForFunctionConversion()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.h181 Status UpdateEdge(TransposeContext* context, absl::string_view name_format,
Dgeneric_layout_optimizer_transposer.cc382 UpdateEdge(context, in UpdateFaninEdgesWithOp()
425 TF_RETURN_IF_ERROR(UpdateEdge( in UpdateFanoutEdgesWithOp()
483 Status Transposer::UpdateEdge( in UpdateEdge() function in tensorflow::grappler::Transposer
/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt271 tensorflow::UpdateEdge
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc565 graph->UpdateEdge(engine_node, conn.port_number, output_node, port)); in CreateTRTNode()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc783 tensorflow::UpdateEdge(graph, new_src, dst, status.get()); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/framework/
Dops.py2409 pywrap_tf_session.UpdateEdge(
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc750 TF_RETURN_IF_ERROR(graph_->UpdateEdge(placeholder_node, 0, edge->dst(), in CreatePlaceholderNodeForFeed()