Home
last modified time | relevance | path

Searched defs:output_edges (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dpython_compiled_autograd.cpp471 const edge_list& output_edges, in _compiled_autograd_impl()
710 const edge_list& output_edges) { in compiled_autograd()
/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DINode.cpp133 const std::set<EdgeID> &INode::output_edges() const in output_edges() function in arm_compute::graph::INode
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dautograd.cpp112 edge_list output_edges; in run_backward() local
H A Dpython_engine.cpp318 std::vector<Edge> output_edges; in THPEngine_run_backward() local
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DInPlaceOperationMutator.cpp56 const auto output_edges = parent_node->output_edges(); in output_edges_are_separate_tensors() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgraph_view.h119 gtl::ArraySlice<EdgeInfo> output_edges() const { in output_edges() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Drearrange_function_argument.cc135 std::vector<const Edge*> output_edges; in ReorderOutputEdges() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc282 std::vector<const Edge*> output_edges; in RemoveIdentityNodesForArgRetval() local
1395 std::vector<std::vector<const Edge*>> output_edges(num_replicas * in RemoveUnusedXlaOutput() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc888 std::map<int, const Edge*> output_edges; in IdentifyAllFunctionsToExecute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc345 std::vector<const Edge*> output_edges; in RemoveDescendantNodeOfArg() local