| /external/ComputeLibrary/src/graph/mutators/ |
| D | NodeFusionMutator.cpp | 104 const auto conv_input_id = conv_node->input_edge(0)->producer_id(); in fuse_convolution_with_batch_normalization()
105 const auto conv_weights_id = conv_node->input_edge(1)->producer_id(); in fuse_convolution_with_batch_normalization()
113 const auto bn_mean_id = bn_node->input_edge(1)->producer_id(); in fuse_convolution_with_batch_normalization()
114 const auto bn_var_id = bn_node->input_edge(2)->producer_id(); in fuse_convolution_with_batch_normalization()
121 if(conv_node->input_edge(2) != nullptr) in fuse_convolution_with_batch_normalization()
123 auto conv_bias_id = conv_node->input_edge(2)->producer_id(); in fuse_convolution_with_batch_normalization()
133 if(bn_node->input_edge(3) != nullptr) in fuse_convolution_with_batch_normalization()
135 const auto bn_beta_id = bn_node->input_edge(3)->producer_id(); in fuse_convolution_with_batch_normalization()
139 if(bn_node->input_edge(4) != nullptr) in fuse_convolution_with_batch_normalization()
141 const auto bn_gamma_id = bn_node->input_edge(4)->producer_id(); in fuse_convolution_with_batch_normalization()
[all …]
|
| D | InPlaceOperationMutator.cpp | 45 bool output_edges_are_separate_tensors(Graph &g, const Edge *input_edge) in output_edges_are_separate_tensors() argument
47 const auto parent_node = input_edge->producer(); in output_edges_are_separate_tensors()
48 const auto input_tensor = input_edge->tensor(); in output_edges_are_separate_tensors()
49 const auto input_edge_id = input_edge->id(); in output_edges_are_separate_tensors()
95 Edge *input_edge = node->input_edge(0); in try_in_place_depthwiseconv() local
96 Edge *weight_edge = node->input_edge(1); in try_in_place_depthwiseconv()
97 ARM_COMPUTE_ERROR_ON(input_edge == nullptr || weight_edge == nullptr); in try_in_place_depthwiseconv()
99 auto input_tensor = input_edge->tensor(); in try_in_place_depthwiseconv()
158 Edge *input0_edge = node->input_edge(0); in try_in_place_elementwise()
159 Edge *input1_edge = node->input_edge(1); in try_in_place_elementwise()
[all …]
|
| D | GroupedConvolutionMutator.cpp | 151 … ARM_COMPUTE_ERROR_ON(conv_node->input_edge(0) == nullptr || conv_node->input_edge(1) == nullptr); in mutate()
152 const NodeID input_id = conv_node->input_edge(0)->producer()->id(); in mutate()
153 const NodeID weights_id = conv_node->input_edge(1)->producer()->id(); in mutate()
154 …const NodeID bias_id = (conv_node->input_edge(2) != nullptr) ? conv_node->input_edge(2)->produc… in mutate()
|
| D | SyntheticDataTypeMutator.cpp | 96 Edge *input_edge = node->input_edge(0); in remove_optimized_nodes() local
97 ARM_COMPUTE_ERROR_ON(input_edge == nullptr); in remove_optimized_nodes()
100 INode *producer = input_edge->producer(); in remove_optimized_nodes()
101 const EdgeID producer_edge_id = input_edge->producer_idx(); in remove_optimized_nodes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | parallel_concat_optimizer.cc | 78 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()
|
| D | shape_refiner.cc | 414 const Edge* input_edge; in EvaluateConstantTensorForEdge() local
415 TF_RETURN_IF_ERROR(node->input_edge(dst_idx, &input_edge)); in EvaluateConstantTensorForEdge()
416 OutputTensor tensor(input_edge->src(), input_edge->src_output()); in EvaluateConstantTensorForEdge()
452 const Edge* input_edge; in ConstantPartialShape() local
453 TF_RETURN_IF_ERROR(node->input_edge(dst_idx, &input_edge)); in ConstantPartialShape()
455 InferenceContext* src_context = GetContext(input_edge->src()); in ConstantPartialShape()
457 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape()
491 const string& src_op = input_edge->src()->type_string(); in ConstantPartialShape()
512 if (!ConstantPartialShape(target_context, input_edge->src(), 0, in ConstantPartialShape()
523 auto* dest_type = input_edge->src()->attrs().Find("DstT"); in ConstantPartialShape()
[all …]
|
| D | accumulate_n_optimizer.cc | 132 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()
|
| D | forward_type_inference.cc | 200 n->input_edge(rev_idx_and_fn.first, &e), in Run()
297 TF_RETURN_IF_ERROR(n->input_edge(reverse.at(nid).first, &e)); in Run()
|
| D | eval_const_tensor.cc | 54 TF_RETURN_IF_ERROR(node.input_edge(input_idx, &edge)); in EvaluateConstantIntFromScalarEdge()
135 TF_RETURN_IF_ERROR(node.input_edge(0, &edge)); in TryToInferTensorOutputFromStridedSliceNode()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | rearrange_function_argument.cc | 152 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 …]
|
| D | tf2xla_util.cc | 121 TF_RETURN_IF_ERROR(replace_node->input_edge(dst_input, &usage_edge)); in ReplaceEdge()
370 TF_RETURN_IF_ERROR(e->src()->input_edge(e->src_output(), &e)); in TraverseUnmodifiedPathBackward()
383 TF_RETURN_IF_ERROR(loop_body->ret_nodes[index]->input_edge(0, &e)); in IsLoopInvariant()
433 const Edge* input_edge; in PropagateConstIntoAndAroundWhileNode() local
434 TF_RETURN_IF_ERROR(while_node->input_edge(i, &input_edge)); in PropagateConstIntoAndAroundWhileNode()
435 TF_ASSIGN_OR_RETURN(input_edge, TraverseUnmodifiedPathBackward( in PropagateConstIntoAndAroundWhileNode()
436 input_edge, lookup_fld, fld, &cache)); in PropagateConstIntoAndAroundWhileNode()
437 if (!input_edge->src()->IsConstant()) { in PropagateConstIntoAndAroundWhileNode()
439 << input_edge->src()->type_string(); in PropagateConstIntoAndAroundWhileNode()
452 const_input_index_to_mutable_node[i] = input_edge->src(); in PropagateConstIntoAndAroundWhileNode()
[all …]
|
| D | resource_util.cc | 157 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()
|
| D | const_analysis.cc | 240 status = pred->src()->input_edge(pred->src_output(), &pred); in BackwardsConstAnalysis()
272 status = edge->src()->input_edge(edge->src_output(), &edge); in BackwardsConstAnalysis()
|
| D | functionalize_cond.cc | 124 TF_RETURN_IF_ERROR(switch_node.input_edge(1, &pred_edge)); in GetSwitchPredicate()
128 TF_RETURN_IF_ERROR(pred_edge->src()->input_edge(0, &pred_edge)); in GetSwitchPredicate()
136 TF_RETURN_IF_ERROR(switch_node.input_edge(0, &val_edge)); in GetSwitchValue()
428 TF_RETURN_IF_ERROR(switch_node->input_edge(0, &e)); in BuildArgumentNodes()
780 TF_RETURN_IF_ERROR(arg->input_edge(0, &in_edge)); in BuildIfNode()
1267 TF_RETURN_IF_ERROR(node->input_edge(0, &e)); in RemoveRedundantSwitch()
1270 TF_RETURN_IF_ERROR(val.node->input_edge(0, &e)); in RemoveRedundantSwitch()
1281 TF_RETURN_IF_ERROR(node->input_edge(0, &value_edge)); in RemoveRedundantSwitch()
|
| /external/ComputeLibrary/src/graph/detail/ |
| D | CrossLayerMemoryManagerHelpers.cpp | 133 Edge *input_edge = node.input_edge(i); in get_transition_handles() local
135 …if(input_edge != nullptr && input_edge->tensor() != nullptr && const_tensors.find(input_edge->tens… in get_transition_handles()
138 ITensorHandle *tensor_handle = input_edge->tensor()->handle()->parent_handle(); in get_transition_handles()
|
| /external/ComputeLibrary/src/graph/ |
| D | Utils.cpp | 206 auto input_edge = g->edge(input_edge_id); in get_driver_nodes() local
207 if(input_edge != nullptr) in get_driver_nodes()
209 ARM_COMPUTE_ERROR_ON(input_edge->producer() == nullptr); in get_driver_nodes()
210 driver_nodes.push_back({ input_edge->producer_id(), input_edge->producer_idx() }); in get_driver_nodes()
|
| D | INode.cpp | 172 Edge *INode::input_edge(size_t idx) const in input_edge() function in arm_compute::graph::INode
|
| /external/ComputeLibrary/src/graph/algorithms/ |
| D | TopologicalSort.cpp | 57 const Edge *input_edge = graph->edge(input_edge_id); in all_inputs_are_visited() local
58 ARM_COMPUTE_ERROR_ON(input_edge == nullptr); in all_inputs_are_visited()
59 ARM_COMPUTE_ERROR_ON(input_edge->producer() == nullptr); in all_inputs_are_visited()
60 if(!visited[input_edge->producer_id()]) in all_inputs_are_visited()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 66 bool input_edge, int port) in EngineConnection()
73 is_input_edge(input_edge), in EngineConnection()
78 int in_id, bool input_edge) in EngineConnection()
85 is_input_edge(input_edge), in EngineConnection()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 279 const Edge* input_edge; in RemoveIdentityNodesForArgRetval() local
280 TF_RETURN_IF_ERROR(n->input_edge(0, &input_edge)); in RemoveIdentityNodesForArgRetval()
290 g->AddControlEdge(input_edge->src(), dst); in RemoveIdentityNodesForArgRetval()
295 g->AddEdge(input_edge->src(), input_edge->src_output(), dst, dst_input); in RemoveIdentityNodesForArgRetval()
689 const Edge* input_edge = in MoveHeadOutsideCompilationToHost() local
695 input_edge->src()->output_type(input_edge->src_output()); in MoveHeadOutsideCompilationToHost()
699 NodeDefBuilder::NodeOut{input_edge->src()->name(), in MoveHeadOutsideCompilationToHost()
700 input_edge->src_output(), input_dtype}); in MoveHeadOutsideCompilationToHost()
708 g->AddEdge(input_edge->src(), input_edge->src_output(), id_node, i); in MoveHeadOutsideCompilationToHost()
1108 Status s = output.first->input_edge(output.second, &original_edge); in MoveTailOutsideCompilationToHost()
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.cc | 269 for (const Edge* input_edge : candidate_edge->src()->out_edges()) { in CreateInputProxy() local
270 if (!input_edge->IsControlEdge() && in CreateInputProxy()
271 input_edge->dst()->attrs().Find(kTpuReplicateAttr) != nullptr) in CreateInputProxy()
272 edges_to_replace.push_back(input_edge); in CreateInputProxy()
285 for (const Edge* input_edge : input_identity_node->in_edges()) in CreateInputProxy() local
286 if (input_edge->src() == candidate_edge->src()) { in CreateInputProxy()
287 *tpu_input_edge = input_edge; in CreateInputProxy()
291 for (const Edge* input_edge : edges_to_replace) { in CreateInputProxy() local
292 graph->RemoveEdge(input_edge); in CreateInputProxy()
293 graph->AddEdge(input_identity_node, 0, input_edge->dst(), in CreateInputProxy()
[all …]
|
| /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 | 145 TF_RETURN_IF_ERROR(dst->input_edge(edges[i].dst_input, &e)); in PreprocessDataEdgesBetweenOutsideCompilations()
248 TF_RETURN_IF_ERROR(replace_node->input_edge(dst_input, &edge_to_replace)); in PostprocessDataEdgesBetweenOutsideCompilations()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_test.cc | 229 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()
|
| /external/ComputeLibrary/arm_compute/graph/ |
| D | INode.h | 218 Edge *input_edge(size_t idx) const;
|