| /aosp_15_r20/external/ComputeLibrary/src/graph/mutators/ |
| H A D | InPlaceOperationMutator.cpp | 45 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
|
| H A D | NodeFusionMutator.cpp | 271 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
|
| H A D | SyntheticDataTypeMutator.cpp | 96 Edge *input_edge = node->input_edge(0); in remove_optimized_nodes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | rearrange_function_argument.cc | 152 const Edge* input_edge; in ReorderOutputEdges() local
246 const Edge* input_edge; in RearrangeOutputEdges() local
489 const Edge* input_edge; in MaybeRewriteIfNode() local
|
| H A D | tf2xla_util.cc | 433 const Edge* input_edge; in PropagateConstIntoAndAroundWhileNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_functional_ops.cc | 269 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | shape_refiner.cc | 414 const Edge* input_edge; in EvaluateConstantTensorForEdge() local
452 const Edge* input_edge; in ConstantPartialShape() local
|
| H A D | parallel_concat_optimizer.cc | 78 for (const Edge* input_edge : n->in_edges()) { in Run() local
|
| H A D | accumulate_n_optimizer.cc | 132 for (const Edge* input_edge : n->in_edges()) { in RewriteIntoTempVariable() local
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/algorithms/ |
| H A D | TopologicalSort.cpp | 57 const Edge *input_edge = graph->edge(input_edge_id); in all_inputs_are_visited() local
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/ |
| H A D | INode.cpp | 172 Edge *INode::input_edge(size_t idx) const in input_edge() function in arm_compute::graph::INode
|
| H A D | Utils.cpp | 206 auto input_edge = g->edge(input_edge_id); in get_driver_nodes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | encapsulate_tpu_computations_pass.cc | 279 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
|
| H A D | distributed_tpu_rewrite_pass.cc | 4339 const Edge* input_edge; in LowerOutsideCompilationFunctionalNodes() local
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/detail/ |
| H A D | CrossLayerMemoryManagerHelpers.cpp | 133 Edge *input_edge = node.input_edge(i); in get_transition_handles() local
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/ |
| H A D | prepare.py | 135 def _union_input_edge_with( argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph.cc | 268 Status Node::input_edge(int idx, const Edge** e) const { in input_edge() function in tensorflow::Node
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | import_model.cc | 2076 for (const auto* input_edge : in_edges) { in ConvertNode() local
|