Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/common_runtime/
Dparallel_concat_optimizer.cc78 for (const Edge* input_edge : n->in_edges()) { in Run() local
79 if (input_edge->IsControlEdge()) { in Run()
80 g->AddControlEdge(input_edge->src(), start); in Run()
87 .Attr("loc", input_edge->dst_input()) in Run()
89 .Input(input_edge->src(), input_edge->src_output()) in Run()
Dshape_refiner.cc398 const Edge* input_edge; in EvaluateConstantTensorForEdge() local
399 TF_RETURN_IF_ERROR(node->input_edge(dst_idx, &input_edge)); in EvaluateConstantTensorForEdge()
400 OutputTensor tensor(input_edge->src(), input_edge->src_output()); in EvaluateConstantTensorForEdge()
436 const Edge* input_edge; in ConstantPartialShape() local
437 TF_RETURN_IF_ERROR(node->input_edge(dst_idx, &input_edge)); in ConstantPartialShape()
439 InferenceContext* src_context = GetContext(input_edge->src()); in ConstantPartialShape()
441 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape()
470 const string& src_op = input_edge->src()->type_string(); in ConstantPartialShape()
491 if (!ConstantPartialShape(target_context, input_edge->src(), 0, in ConstantPartialShape()
502 auto* dest_type = input_edge->src()->attrs().Find("DstT"); in ConstantPartialShape()
[all …]
Daccumulate_n_optimizer.cc132 for (const Edge* input_edge : n->in_edges()) { in RewriteIntoTempVariable() local
133 if (input_edge->IsControlEdge()) { in RewriteIntoTempVariable()
134 control_edges.push_back(input_edge); in RewriteIntoTempVariable()
136 data_edges.push_back(input_edge); in RewriteIntoTempVariable()
Dplacer_inspection_required_ops_utils.cc171 TF_RETURN_IF_ERROR(node->input_edge(input_idx, &edge)); in AddInputIdentity()
/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc152 const Edge* input_edge; in ReorderOutputEdges() local
153 TF_RETURN_IF_ERROR(n->input_edge(new_src_output, &input_edge)); in ReorderOutputEdges()
154 g->AddEdge(input_edge->src(), input_edge->src_output(), dst, dst_input); in ReorderOutputEdges()
194 TF_RETURN_IF_ERROR(n->input_edge(0, &e)); in CalculateRetvalRearrange()
246 const Edge* input_edge; in RearrangeOutputEdges() local
248 n->input_edge(resource_retval_to_arg.at(src_output), &input_edge)); in RearrangeOutputEdges()
249 g->AddEdge(input_edge->src(), input_edge->src_output(), dst, dst_input); in RearrangeOutputEdges()
489 const Edge* input_edge; in MaybeRewriteIfNode() local
490 TF_RETURN_IF_ERROR(n->input_edge( in MaybeRewriteIfNode()
491 resource_retval_to_arg.at(src_output) + 1, &input_edge)); in MaybeRewriteIfNode()
[all …]
Dresource_util.cc157 TF_RETURN_IF_ERROR(call_node.input_edge(index, &e)); in UpdateResourceUsageFromFunctionBodyAnalysis()
236 TF_RETURN_IF_ERROR(n.input_edge(0, &in)); in PropagateThroughIdentityOp()
244 TF_RETURN_IF_ERROR(n.input_edge(o->src_output(), &in)); in PropagateThroughIdentityOp()
Dconst_analysis.cc236 status = pred->src()->input_edge(pred->src_output(), &pred); in BackwardsConstAnalysis()
266 status = edge->src()->input_edge(edge->src_output(), &edge); in BackwardsConstAnalysis()
Dfunctionalize_cond.cc126 TF_RETURN_IF_ERROR(switch_node.input_edge(1, &pred_edge)); in GetSwitchPredicate()
130 TF_RETURN_IF_ERROR(pred_edge->src()->input_edge(0, &pred_edge)); in GetSwitchPredicate()
138 TF_RETURN_IF_ERROR(switch_node.input_edge(0, &val_edge)); in GetSwitchValue()
430 TF_RETURN_IF_ERROR(switch_node->input_edge(0, &e)); in BuildArgumentNodes()
782 TF_RETURN_IF_ERROR(arg->input_edge(0, &in_edge)); in BuildIfNode()
1274 TF_RETURN_IF_ERROR(node->input_edge(0, &e)); in RemoveRedundantSwitch()
1277 TF_RETURN_IF_ERROR(val.node->input_edge(0, &e)); in RemoveRedundantSwitch()
1288 TF_RETURN_IF_ERROR(node->input_edge(0, &value_edge)); in RemoveRedundantSwitch()
Dfunctionalize_while.cc453 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); in FunctionalizeLoop()
469 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); in FunctionalizeLoop()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc305 const Edge* input_edge; in RemoveIdentityNodesForArgRetval() local
306 TF_RETURN_IF_ERROR(n->input_edge(0, &input_edge)); in RemoveIdentityNodesForArgRetval()
316 g->AddControlEdge(input_edge->src(), dst); in RemoveIdentityNodesForArgRetval()
321 g->AddEdge(input_edge->src(), input_edge->src_output(), dst, dst_input); in RemoveIdentityNodesForArgRetval()
719 const Edge* input_edge = in MoveHeadOutsideCompilationToHost() local
725 input_edge->src()->output_type(input_edge->src_output()); in MoveHeadOutsideCompilationToHost()
729 NodeDefBuilder::NodeOut{input_edge->src()->name(), in MoveHeadOutsideCompilationToHost()
730 input_edge->src_output(), input_dtype}); in MoveHeadOutsideCompilationToHost()
739 g->AddEdge(input_edge->src(), input_edge->src_output(), id_node, i); in MoveHeadOutsideCompilationToHost()
1141 Status s = output.first->input_edge(output.second, &original_edge); in MoveTailOutsideCompilationToHost()
[all …]
Dvariable_merger_pass.cc119 TF_RETURN_IF_ERROR(nodes[i]->input_edge(0, &handle_edge)); in MergeReadVariableOps()
Ddistributed_tpu_rewrite_pass.cc2120 TF_RETURN_IF_ERROR(retvals[i]->input_edge(0, &edge)); in AssignArgsAndRetvalsToCores()
2558 TF_RETURN_IF_ERROR(node->input_edge(i, &e)); in ConnectHostComputeNodes()
2573 TF_RETURN_IF_ERROR(node->input_edge(input_index, &key_edge)); in ConnectHostComputeNodes()
3885 const Edge* input_edge; in LowerOutsideCompilationFunctionalNodes() local
3886 TF_RETURN_IF_ERROR(n->input_edge(src_output, &input_edge)); in LowerOutsideCompilationFunctionalNodes()
3890 g->AddEdge(input_edge->src(), input_edge->src_output(), dst, dst_input); in LowerOutsideCompilationFunctionalNodes()
Dhost_training_loop_optimization_util.cc156 TF_RETURN_IF_ERROR(execute->input_edge(i, &edges[j])); in ExtractExecuteNodeInfo()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h50 bool input_edge, int port) in EngineConnection()
57 is_input_edge(input_edge), in EngineConnection()
62 int in_id, bool input_edge) in EngineConnection()
69 is_input_edge(input_edge), in EngineConnection()
/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass.cc97 TF_RETURN_IF_ERROR(slice->input_edge(kSliceInputIndex, &slice_input_edge)); in GetSliceInputs()
99 TF_RETURN_IF_ERROR(slice->input_edge(kSliceSizeIndex, &slice_size_edge)); in GetSliceInputs()
101 TF_RETURN_IF_ERROR(slice->input_edge(kSliceBeginIndex, &slice_begin_edge)); in GetSliceInputs()
Dencapsulate_util.cc145 TF_RETURN_IF_ERROR(dst->input_edge(edges[i].dst_input, &e)); in PreprocessDataEdgesBetweenOutsideCompilations()
250 TF_RETURN_IF_ERROR(replace_node->input_edge(dst_input, &edge_to_replace)); in PostprocessDataEdgesBetweenOutsideCompilations()
Dnode_matchers.cc306 if (!node->input_edge(input_idx, &edge).ok()) { in MatchAndExplainInput()
/external/tensorflow/tensorflow/core/graph/
Dgraph_test.cc229 TF_ASSERT_OK(c->input_edge(0, &e)); in TEST_F()
247 TF_ASSERT_OK(t->input_edge(1, &e)); in TEST_F()
274 TF_ASSERT_OK(c->input_edge(0, &a_new_c_edge)); in TEST_F()
279 TF_ASSERT_OK(c->input_edge(0, &b_new_c_edge)); in TEST_F()
285 TF_ASSERT_OK(c->input_edge(0, &b_new_c_edge)); in TEST_F()
Dgraph.cc244 Status Node::input_edge(int idx, const Edge** e) const { in input_edge() function in tensorflow::Node
297 TF_RETURN_IF_ERROR(input_edge(idx, &e)); in input_node()
315 TF_RETURN_IF_ERROR(input_edge(idx, &e)); in input_tensor()
Dgraph.h219 Status input_edge(int idx, const Edge** e) const;
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc302 map_defun_fn_->ret_nodes[output_position]->input_edge(0, &ret_edge)); in ConvertOutput()
602 TF_RETURN_IF_ERROR(ret_node->input_edge(0, &in_edge)); in AddUnstackedTensorMappings()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc145 TF_RETURN_IF_ERROR(node->input_edge(0, &in_edge)); in ReplaceBroadcast()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc514 TF_RETURN_IF_ERROR(outputs[r].node->input_edge(0, &edge)); in GraphToFunctionDef()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc732 TF_CHECK_OK(node.input_edge(1, &edge)); in RegisterPadNode()
873 TF_CHECK_OK(node.input_edge(idx, &edge)); in AddNodeInputByInputIndex()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1916 for (const auto* input_edge : in_edges) { in ConvertNode() local
1917 const Node& input_node = *input_edge->src(); in ConvertNode()
1926 if (input_node.IsArg() && input_edge->IsControlEdge()) { in ConvertNode()
1943 if (input_edge->IsControlEdge()) in ConvertNode()
1946 result.operands.push_back(inst->getResult(input_edge->src_output())); in ConvertNode()

12