Home
last modified time | relevance | path

Searched defs:Graph (Results 1 – 25 of 221) sorted by relevance

123456789

/external/tensorflow/tensorflow/go/
Dgraph.go52 type Graph struct { struct
53 c *C.TF_Graph
98 func (g *Graph) finalizer() {
105 func (g *Graph) WriteTo(w io.Writer) (int64, error) {
135 func (g *Graph) ImportWithOptions(def []byte, options GraphImportOptions) error {
178 func (g *Graph) Import(def []byte, prefix string) error {
184 func (g *Graph) Operation(name string) *Operation {
195 func (g *Graph) Operations() []Operation {
213 func (g *Graph) AddGradients(prefix string, y []Output, x []Output, dx []Output) ([]Output, error) {
311 func (g *Graph) AddOperation(args OpSpec) (*Operation, error) {
Dutil_test.go19 func Placeholder(g *Graph, name string, dt DataType) (Output, error) {
30 func Const(g *Graph, name string, value interface{}) (Output, error) {
49 func Neg(g *Graph, name string, port Output) (Output, error) {
58 func Add(g *Graph, name string, x, y Output) (Output, error) {
/external/pytorch/torch/_inductor/fx_passes/
Dddp_fusion.py51 graph: fx.Graph,
162 graph: fx.Graph,
184 graph: fx.Graph,
242 graph: fx.Graph,
304 graph: fx.Graph,
390 graph: fx.Graph,
437 graph: fx.Graph, comm_blocks: List[CommBlock], bucket_size_mb: int
476 graph: fx.Graph, algorithm_fn: Callable[..., Any], fusion_fn: Callable[..., Any]
516 def fuse_ddp_with_coalesced_op(graph: fx.Graph, bucket_size_mb: int) -> None:
524 def fuse_ddp_with_concat_op(graph: fx.Graph, bucket_size_mb: int) -> None:
[all …]
Dmicro_pipeline_tp.py26 def _is_backward(graph: torch.fx.Graph) -> bool:
82 def find_all_gather_patterns(graph: torch.fx.Graph):
218 def find_reduce_scatter_patterns(graph: torch.fx.Graph):
469 graph: torch.fx.Graph,
636 graph: torch.fx.Graph,
752 graph: torch.fx.Graph,
767 graph: torch.fx.Graph,
798 def _get_unexposed_collectives(graph: torch.fx.Graph) -> List[torch.fx.Node]:
834 def micro_pipeline_tp_pass(graph: torch.fx.Graph):
Dreinplace.py45 def graph_call_function(graph: torch.fx.Graph, fn, *args, **kwargs):
92 graph: torch.fx.Graph,
116 graph: torch.fx.Graph, node: torch.fx.Node
134 graph: torch.fx.Graph, node: torch.fx.Node
204 def decompose_generalized_scatter(graph: torch.fx.Graph) -> None:
225 def canonicalize_view_scatter_ops(graph: torch.fx.Graph) -> None:
371 def reinplace_inplaceable_ops_core(graph: torch.fx.Graph) -> None:
685 def reinplace_inplaceable_ops(graph: torch.fx.Graph) -> None:
Dsplit_cat.py608 graph: torch.fx.Graph,
817 graph: torch.fx.Graph,
1043 graph: torch.fx.Graph,
1709 graph: torch.fx.Graph,
1766 graph: torch.fx.Graph,
1802 graph: torch.fx.Graph,
1869 graph: torch.fx.Graph,
1940 def remove_split_unbind_children(graph: torch.fx.Graph, inputs: List[torch.fx.Node]):
2117 graph: torch.fx.Graph,
2170 graph: torch.fx.Graph,
/external/pytorch/torch/fx/
Dsubgraph_rewriter.py209 pattern: Union[Callable, Graph, GraphModule], argument
211 match_filters: Optional[List[Callable[["InternalMatch", Graph, Graph], bool]]] = None, argument
229 pattern: Union[Callable, Graph, GraphModule], argument
231 match_filters: Optional[List[Callable[["InternalMatch", Graph, Graph], bool]]] = None, argument
/external/pytorch/torch/csrc/jit/ir/
Dir.h1180 struct Graph : std::enable_shared_from_this<Graph> { struct
1213 Graph(ScopePtr scope_root = c10::make_intrusive<Scope>()) argument
1214 : current_scope_(std::move(scope_root)), in current_scope_()
1218 at::ArrayRef<Value*> inputs() { in inputs()
1221 at::ArrayRef<const Value*> inputs() const { in inputs()
1225 at::ArrayRef<Value*> outputs() { in outputs()
1228 at::ArrayRef<const Value*> outputs() const { in outputs()
1232 graph_node_list nodes() { in nodes()
1235 const_graph_node_list nodes() const { in nodes()
1239 Node* param_node() { in param_node()
[all …]
/external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/Analysis/
DDOTGraphTraitsPass.h31 void viewGraphForFunction(Function &F, GraphT Graph, StringRef Name, in viewGraphForFunction()
64 GraphT Graph = AnalysisGraphTraitsT::getGraph(Result); in run() local
87 void printGraphForFunction(Function &F, GraphT Graph, StringRef Name, in printGraphForFunction()
130 GraphT Graph = AnalysisGraphTraitsT::getGraph(Result); in run() local
185 GraphT Graph = AnalysisGraphTraitsT::getGraph(&Analysis); in runOnFunction() local
224 GraphT Graph = AnalysisGraphTraitsT::getGraph(&Analysis); in runOnFunction() local
248 GraphT Graph = AnalysisGraphTraitsT::getGraph(&getAnalysis<AnalysisT>()); in runOnModule() local
274 GraphT Graph = AnalysisGraphTraitsT::getGraph(&getAnalysis<AnalysisT>()); in runOnModule() local
302 void WriteDOTGraphToFile(Function &F, GraphT &&Graph, in WriteDOTGraphToFile()
/external/pytorch/torch/fx/passes/utils/
Dcommon.py31 subgraph: Graph,
85 def compare_graphs(left: Graph, right: Graph) -> bool:
/external/executorch/backends/transforms/
Daddmm_mm_to_linear.py60 def replace_linear_view_copy_input_output(graph: torch.fx.Graph) -> torch.fx.Graph:
102 def replace_addmm_mm_with_linear(graph: torch.fx.Graph) -> torch.fx.Graph:
161 def apply_addmm_mm_to_linear_transform(graph: torch.fx.Graph) -> torch.fx.Graph:
/external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/Analysis/
DDOTGraphTraitsPass.h54 GraphT Graph = AnalysisGraphTraitsT::getGraph(&Analysis); in runOnFunction() local
96 GraphT Graph = AnalysisGraphTraitsT::getGraph(&Analysis); in runOnFunction() local
133 GraphT Graph = AnalysisGraphTraitsT::getGraph(&getAnalysis<AnalysisT>()); in runOnModule() local
159 GraphT Graph = AnalysisGraphTraitsT::getGraph(&getAnalysis<AnalysisT>()); in runOnModule() local
/external/llvm/include/llvm/Analysis/
DDOTGraphTraitsPass.h55 GraphT Graph = AnalysisGraphTraitsT::getGraph(&Analysis); in runOnFunction() local
97 GraphT Graph = AnalysisGraphTraitsT::getGraph(&Analysis); in runOnFunction() local
134 GraphT Graph = AnalysisGraphTraitsT::getGraph(&getAnalysis<AnalysisT>()); in runOnModule() local
160 GraphT Graph = AnalysisGraphTraitsT::getGraph(&getAnalysis<AnalysisT>()); in runOnModule() local
/external/executorch/exir/
Dprint_program.py317 def add_cursor_to_graph(graph: torch.fx.Graph, finding_node: torch.fx.Node) -> str:
340 def _format_graph(graph: torch.fx.Graph, offending_node_idx: int) -> str:
370 def inspect_node(graph: torch.fx.Graph, node: torch.fx.Node) -> str:
/external/executorch/backends/vulkan/_passes/
Dremove_local_scalar_dense_ops.py58 def remove_local_scalar_dense_chain(graph: torch.fx.Graph, node: torch.fx.Node) -> None:
83 def remove_local_scalar_dense_ops(graph: torch.fx.Graph) -> torch.fx.Graph:
/external/pytorch/torch/ao/quantization/fx/
Dgraph_module.py22 graph: Graph,
52 graph: Graph,
105 graph: Graph,
152 graph: Graph,
Dconvert.py500 node: Node, graph: Graph
560 def _maybe_recursive_remove_dequantize(arg: Any, node: Node, graph: Graph) -> None:
638 def _insert_dequantize_node(node: Node, graph: Graph) -> None:
871 node: Node, prev_node: Node, graph: Graph
893 graph: Graph,
/external/executorch/backends/mediatek/quantizer/
Dannotator.py31 def annotate(graph: Graph, quant_config: QuantizationConfig) -> None:
84 graph: Graph, quant_config: QuantizationConfig
149 def _annotate_rmsnorm_pattern(graph: Graph, quant_config: QuantizationConfig) -> None:
/external/executorch/backends/xnnpack/_passes/
Dconv1d_unsqueeze_pass.py30 graph: torch.fx.Graph,
44 graph: torch.fx.Graph,
/external/executorch/backends/arm/_passes/
Darm_pass_utils.py67 graph: torch.fx.Graph,
91 graph: torch.fx.Graph,
Dsize_adjust_conv2d_pass.py26 graph: torch.fx.Graph,
54 graph: torch.fx.Graph,
/external/rust/android-crates-io/crates/petgraph/src/graph_impl/
Dmod.rs347 pub struct Graph<N, E, Ty = Directed, Ix = DefaultIx> { struct
353 /// A `Graph` with directed edges. argument
366 impl<N, E, Ty, Ix: IndexType> Clone for Graph<N, E, Ty, Ix> implementation
386 impl<N, E, Ty, Ix> fmt::Debug for Graph<N, E, Ty, Ix> implementation
455 impl<N, E> Graph<N, E, Directed> { impl
469 impl<N, E> Graph<N, E, Undirected> { impl
483 impl<N, E, Ty, Ix> Graph<N, E, Ty, Ix> impl
1611 impl<'a, N, E, Ty, Ix> visit::IntoEdges for &'a Graph<N, E, Ty, Ix> implementation
1622 impl<'a, N, E, Ty, Ix> visit::IntoEdgesDirected for &'a Graph<N, E, Ty, Ix> implementation
1853 impl<N, E, Ty, Ix> Index<NodeIndex<Ix>> for Graph<N, E, Ty, Ix> implementation
[all …]
/external/pytorch/torch/csrc/jit/codegen/cuda/
DREADME.md43 ##### Usage
55 ##### Example Output
/external/executorch/backends/cadence/aot/
Dcompiler_utils.py20 def get_output_node(graph: torch.fx.Graph) -> torch.fx.Node:
30 def is_node_in_flattened_output(graph: torch.fx.Graph, node: torch.fx.Node) -> bool:
36 def get_placeholders(graph: torch.fx.Graph) -> List[torch.fx.Node]:
/external/executorch/build/
Dextract_sources.py100 self, graph: "Graph", runner: Buck2Runner, buck_args: Optional[List[str]]
145 class Graph: class

123456789