Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/graph/mutators/
DInPlaceOperationMutator.cpp45 bool output_edges_are_separate_tensors(Graph &g, const Edge *input_edge) in output_edges_are_separate_tensors()
95 Edge *input_edge = node->input_edge(0); in try_in_place_depthwiseconv() local
233 Edge *input_edge = node->input_edge(0); in mutate() local
DNodeFusionMutator.cpp271 const Edge *input_edge = pad_node->input_edge(0); in fuse_pad_with_convolution() local
559 const Edge *input_edge = conv_node->input_edge(1); in fuse_convolution_with_post_ops() local
657 const Edge *input_edge = conv_node->input_edge(1); in fuse_convolution_batch_normalization_with_post_ops() 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.cc414 const Edge* input_edge; in EvaluateConstantTensorForEdge() local
452 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.cc433 const Edge* input_edge; in PropagateConstIntoAndAroundWhileNode() local
/external/ComputeLibrary/src/graph/
DINode.cpp172 Edge *INode::input_edge(size_t idx) const in input_edge() function in arm_compute::graph::INode
DUtils.cpp206 auto input_edge = g->edge(input_edge_id); in get_driver_nodes() local
/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.cc269 for (const Edge* input_edge : candidate_edge->src()->out_edges()) { in CreateInputProxy() local
285 for (const Edge* input_edge : input_identity_node->in_edges()) in CreateInputProxy() local
291 for (const Edge* input_edge : edges_to_replace) { in CreateInputProxy() local
344 const Edge* input_edge = nullptr; in RemoveDescendantNodeOfArg() local
565 for (const Edge* input_edge : tpu_replicated_input_node->in_edges()) { in FindTpuReplicatedInputAndXlaSharding() local
571 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.cc279 const Edge* input_edge; in RemoveIdentityNodesForArgRetval() local
689 const Edge* input_edge = in MoveHeadOutsideCompilationToHost() local
1492 const Edge* input_edge = input_edges.at(input_index); in RemoveEdgesBetweenArgAndRetval() local
2033 const Edge* input_edge; in ReplaceOutputEdgesWithInputEdgeSourceForWhile() local
Ddistributed_tpu_rewrite_pass.cc4339 const Edge* input_edge; in LowerOutsideCompilationFunctionalNodes() local
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc268 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.cc2076 for (const auto* input_edge : in_edges) { in ConvertNode() local