Home
last modified time | relevance | path

Searched defs:input_edge (Results 1 – 15 of 15) sorted by relevance

/external/ComputeLibrary/src/graph/mutators/
DInPlaceOperationMutator.cpp38 bool output_edges_are_separate_tensors(Graph &g, const Edge *input_edge) in output_edges_are_separate_tensors()
101 Edge *input_edge = node->input_edge(0); in mutate() local
DSyntheticDataTypeMutator.cpp96 Edge *input_edge = node->input_edge(0); in remove_optimized_nodes() local
/external/tensorflow/tensorflow/core/common_runtime/
Dparallel_concat_optimizer.cc78 for (const Edge* input_edge : n->in_edges()) { in Run() local
Daccumulate_n_optimizer.cc132 for (const Edge* input_edge : n->in_edges()) { in RewriteIntoTempVariable() local
Dshape_refiner.cc415 const Edge* input_edge; in EvaluateConstantTensorForEdge() local
453 const Edge* input_edge; in ConstantPartialShape() local
/external/ComputeLibrary/src/graph/algorithms/
DTopologicalSort.cpp57 const Edge *input_edge = graph->edge(input_edge_id); in all_inputs_are_visited() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc152 const Edge* input_edge; in ReorderOutputEdges() local
246 const Edge* input_edge; in RearrangeOutputEdges() local
489 const Edge* input_edge; in MaybeRewriteIfNode() local
Dtf2xla_util.cc435 const Edge* input_edge; in PropagateConstIntoAndAroundWhileNode() local
/external/ComputeLibrary/src/graph/
DINode.cpp171 Edge *INode::input_edge(size_t idx) const in input_edge() function in arm_compute::graph::INode
/external/ComputeLibrary/src/graph/detail/
DCrossLayerMemoryManagerHelpers.cpp133 Edge *input_edge = node.input_edge(i); in get_transition_handles() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_functional_ops.cc253 for (const Edge* input_edge : candidate_edge->src()->out_edges()) { in CreateInputProxy() local
269 for (const Edge* input_edge : input_identity_node->in_edges()) in CreateInputProxy() local
275 for (const Edge* input_edge : edges_to_replace) { in CreateInputProxy() local
328 const Edge* input_edge = nullptr; in RemoveDescendantNodeOfArg() local
549 for (const Edge* input_edge : tpu_replicated_input_node->in_edges()) { in FindTpuReplicatedInputAndXlaSharding() local
555 for (const Edge* input_edge : tpu_replicated_input_node->in_edges()) { in FindTpuReplicatedInputAndXlaSharding() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc305 const Edge* input_edge; in RemoveIdentityNodesForArgRetval() local
719 const Edge* input_edge = in MoveHeadOutsideCompilationToHost() local
1533 const Edge* input_edge = input_edges.at(input_index); in RemoveEdgesBetweenArgAndRetval() local
2079 const Edge* input_edge; in ReplaceOutputEdgesWithInputEdgeSourceForWhile() local
Ddistributed_tpu_rewrite_pass.cc4392 const Edge* input_edge; in LowerOutsideCompilationFunctionalNodes() local
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc249 Status Node::input_edge(int idx, const Edge** e) const { in input_edge() function in tensorflow::Node
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc2011 for (const auto* input_edge : in_edges) { in ConvertNode() local