Home
last modified time | relevance | path

Searched defs:output_edge (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DNodeFusionMutator.cpp82 void fuse_convolution_with_batch_normalization(Graph &g, const Edge *output_edge) in fuse_convolution_with_batch_normalization()
162 void fuse_depthwise_convolution_with_batch_normalization(Graph &g, const Edge *output_edge) in fuse_depthwise_convolution_with_batch_normalization()
227 void fuse_node_with_activation(Graph &g, const Edge *output_edge, const std::set<Activation> &suppo… in fuse_node_with_activation()
266 void fuse_pad_with_convolution(Graph &g, const Edge *output_edge) in fuse_pad_with_convolution()
324 const auto output_edge = g.edge(output_edge_id); in fuse_layer() local
541 void fuse_convolution_with_post_ops(Graph &g, const Edge *output_edge, unsigned int conv_node_id, c… in fuse_convolution_with_post_ops()
640 void fuse_convolution_batch_normalization_with_post_ops(Graph &g, const Edge *output_edge, unsigned… in fuse_convolution_batch_normalization_with_post_ops()
783 const auto output_edge = g.edge(output_edge_id); in fuse_layer() local
829 const auto output_edge = g.edge(output_edge_id); in mutate() local
/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DINode.cpp88 auto output_edge = _graph->edge(output_edge_id); in set_output_tensor() local
H A DUtils.cpp186 auto output_edge = g->edge(output_edge_id); in get_driving_nodes() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dengine.cpp1143 for (auto& output_edge : outputs) { in compute_min_topological_nr() local
1595 for (auto& output_edge : outputs) { in init_to_execute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc359 for (const Edge* output_edge : it.second) { in RemoveDescendantNodeOfArg() local
1040 for (const Edge* output_edge : edge->dst()->out_edges()) in CreateConcatAndSplitNodesForOutputTensor() local
1049 for (const Edge* output_edge : output_edge_vec) { in CreateConcatAndSplitNodesForOutputTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dhost_training_loop_optimization_util.cc96 for (const Edge* output_edge : enter_node->out_edges()) { in ResourceOnlyUsedForTPUExecuteInLoop() local
H A Dencapsulate_tpu_computations_pass.cc1495 for (const Edge* output_edge : xla_node_out_edges[output_index]) { in RemoveEdgesBetweenArgAndRetval() local