| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | parallel_concat_optimizer.cc | 83 for (const Edge* input_edge : n->in_edges()) { in Run() local
84 if (input_edge->IsControlEdge()) { in Run()
85 g->AddControlEdge(input_edge->src(), start); in Run()
94 .Input(input_edge->src(), input_edge->src_output()) in Run()
|
| D | shape_refiner.cc | 420 const Edge* input_edge; in EvaluateConstantTensorForEdge() local
421 TF_RETURN_IF_ERROR(node->input_edge(dst_idx, &input_edge)); in EvaluateConstantTensorForEdge()
422 OutputTensor tensor(input_edge->src(), input_edge->src_output()); in EvaluateConstantTensorForEdge()
454 const Edge* input_edge; in ConstantPartialShape() local
455 TF_RETURN_IF_ERROR(node->input_edge(dst_idx, &input_edge)); in ConstantPartialShape()
457 InferenceContext* src_context = GetContext(input_edge->src()); in ConstantPartialShape()
459 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape()
488 const string& src_op = input_edge->src()->type_string(); in ConstantPartialShape()
496 *result = src_context->input(input_edge->src_output()); in ConstantPartialShape()
503 TF_RETURN_IF_ERROR(EvaluateConstantIntScalarEdge(input_edge->src(), i, in ConstantPartialShape()
[all …]
|
| D | accumulate_n_optimizer.cc | 100 for (const Edge* input_edge : n->in_edges()) { in rewriteNode() local
101 if (input_edge->IsControlEdge()) { in rewriteNode()
102 control_edges.push_back(input_edge); in rewriteNode()
104 data_edges.push_back(input_edge); in rewriteNode()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 56 bool input_edge, int port) in EngineConnection()
63 is_input_edge(input_edge), in EngineConnection()
68 int in_id, bool input_edge) in EngineConnection()
75 is_input_edge(input_edge), in EngineConnection()
|
| D | convert_graph.cc | 78 for (const Edge* input_edge : input_edges) { in IsTensorRTCandidate() local
79 input_node_and_ports.emplace_back(&input_edge->src()->def(), in IsTensorRTCandidate()
80 input_edge->src_output()); in IsTensorRTCandidate()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 81 status = pred->src()->input_edge(pred->src_output(), &pred); in BackwardsConstAnalysis()
111 status = edge->src()->input_edge(edge->src_output(), &edge); in BackwardsConstAnalysis()
|
| D | functionalize_cond.cc | 122 TF_RETURN_IF_ERROR(switch_node.input_edge(1, &pred_edge)); in GetSwitchPredicate()
126 TF_RETURN_IF_ERROR(pred_edge->src()->input_edge(0, &pred_edge)); in GetSwitchPredicate()
134 TF_RETURN_IF_ERROR(switch_node.input_edge(0, &val_edge)); in GetSwitchValue()
414 TF_RETURN_IF_ERROR(switch_node->input_edge(0, &e)); in BuildArgumentNodes()
756 TF_RETURN_IF_ERROR(arg->input_edge(0, &in_edge)); in BuildIfNode()
1227 TF_RETURN_IF_ERROR(node->input_edge(0, &e)); in RemoveRedundantSwitch()
1230 TF_RETURN_IF_ERROR(val.node->input_edge(0, &e)); in RemoveRedundantSwitch()
1241 TF_RETURN_IF_ERROR(node->input_edge(0, &value_edge)); in RemoveRedundantSwitch()
|
| D | functionalize_while.cc | 535 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); in FunctionalizeLoop()
551 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge)); in FunctionalizeLoop()
|
| D | tf2xla_util.cc | 155 replace_node->input_edge(usages[i].dst_input, &usage_edge)); in ReplaceArgUsageWithConstNode()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass.cc | 97 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()
|
| D | encapsulate_util.cc | 139 TF_RETURN_IF_ERROR(dst->input_edge(edges[i].dst_input, &e)); in PreprocessDataEdgesBetweenOutsideCompilations()
244 TF_RETURN_IF_ERROR(replace_node->input_edge(dst_input, &edge_to_replace)); in PostprocessDataEdgesBetweenOutsideCompilations()
|
| D | node_matchers.cc | 302 if (!node->input_edge(input_idx, &edge).ok()) { in MatchAndExplainInput()
|
| D | deadness_analysis.cc | 792 TF_RETURN_IF_ERROR(n->input_edge(1, &pred_edge)); in HandleSwitch()
|
| D | extract_outside_compilation_pass.cc | 270 if (!send_from_host_node->input_edge(i, &e).ok()) { in GetInferredInputShapes()
|
| D | encapsulate_subgraphs_pass.cc | 2059 !dst->input_edge(0, &existing_input_edge).ok()) { in DoStaticShapeInferenceForOutsideCompilationSend()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_test.cc | 228 TF_ASSERT_OK(c->input_edge(0, &e)); in TEST_F()
246 TF_ASSERT_OK(t->input_edge(1, &e)); in TEST_F()
273 TF_ASSERT_OK(c->input_edge(0, &a_new_c_edge)); in TEST_F()
278 TF_ASSERT_OK(c->input_edge(0, &b_new_c_edge)); in TEST_F()
284 TF_ASSERT_OK(c->input_edge(0, &b_new_c_edge)); in TEST_F()
|
| D | graph.cc | 238 Status Node::input_edge(int idx, const Edge** e) const { in input_edge() function in tensorflow::Node
291 TF_RETURN_IF_ERROR(input_edge(idx, &e)); in input_node()
309 TF_RETURN_IF_ERROR(input_edge(idx, &e)); in input_tensor()
|
| D | graph.h | 191 Status input_edge(int idx, const Edge** e) const;
|
| D | mkl_layout_pass.cc | 3528 TF_CHECK_OK(n->input_edge(idx, &e)); in FixMklMetaDataEdges()
3552 TF_CHECK_OK(n->input_edge(e_meta_in_slot, &e_meta)); in FixMklMetaDataEdges()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 302 map_defun_fn_->ret_nodes[output_position]->input_edge(0, &ret_edge)); in ConvertOutput()
609 TF_RETURN_IF_ERROR(ret_node->input_edge(0, &in_edge)); in AddUnstackedTensorMappings()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.cc | 156 TF_RETURN_IF_ERROR(node->input_edge(0, &edge)); in GraphToFunctionDef()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 145 TF_RETURN_IF_ERROR(node->input_edge(0, &in_edge)); in ReplaceBroadcast()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 732 TF_CHECK_OK(node.input_edge(1, &edge)); in RegisterPadNode()
873 TF_CHECK_OK(node.input_edge(idx, &edge)); in AddNodeInputByInputIndex()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 973 TF_RETURN_IF_ERROR(node->input_edge(i, &edge)); in FuseCluster()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 1560 Status s = oper_in.oper->node.input_edge(oper_in.index, &edge); in TF_OperationInput()
|