| /aosp_15_r20/external/executorch/backends/cadence/aot/ |
| H A D | remove_ops.py | 549 class Subgraph: class in RemovePermutesAroundElementwiseOps
606 def is_valid_subgraph(self, sg: Subgraph) -> bool:
614 def handle_subgraph(self, graph_module: torch.fx.GraphModule, sg: Subgraph) -> None:
625 def start_search(self, node: torch.fx.Node, sg: Subgraph) -> None:
635 def search_up(self, node: object, sg: Subgraph) -> None:
665 def search_down(self, node: torch.fx.Node, sg: Subgraph) -> None:
682 def traverse_intermediate_node(self, node: torch.fx.Node, sg: Subgraph) -> None:
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| H A D | raise_target_subgraphs.cc | 61 struct Subgraph { struct
63 llvm::SetVector<Operation*> all_ops;
64 int subgraph_id;
65 InferenceDeviceType inference_device_type;
|
| H A D | pick_subgraphs.cc | 96 struct Subgraph { struct
98 func::CallOp call;
104 available_choices;
112 aggregated_cost_with_decisions;
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | b2b_gemm.py | 425 subgraph: Subgraph,
447 subgraph: Subgraph,
515 subgraph: Subgraph,
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/ |
| H A D | splitter_base.py | 199 class Subgraph: class
826 def remove_small_acc_subgraphs(self, subgraphs: List[Subgraph]) -> List[Subgraph]:
853 def tag(self, subgraphs: List[Subgraph]):
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/include/nnapi/ |
| D | Types.h | 716 struct Subgraph { struct
720 std::vector<Operand> operands;
729 std::vector<Operation> operations;
736 std::vector<uint32_t> inputIndexes;
743 std::vector<uint32_t> outputIndexes;
|
| /aosp_15_r20/out/soong/.intermediates/hardware/interfaces/neuralnetworks/aidl/android.hardware.neuralnetworks-V4-ndk-source/gen/include/aidl/android/hardware/neuralnetworks/ |
| D | Model.h | 28 class Subgraph; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| H A D | subgraph.h | 49 explicit Subgraph(Identity id) : id_(std::move(id)), hash_(id_.Hash()) {} in Subgraph() function
|
| H A D | subgraph.cc | 112 Subgraph::Subgraph(const Identity& parent_id, GenNode* add_node) in Subgraph() function in tensorflow::grappler::graph_analyzer::Subgraph
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | subgraph_lowering.py | 105 subgraph: ir.Subgraph, inputs: List[InputDescriptor]
|
| H A D | ir.py | 7598 class Subgraph(IRNode): class
7625 true_subgraph: Subgraph,
7626 false_subgraph: Subgraph,
7651 true_fn: Subgraph,
7652 false_fn: Subgraph,
7745 cond_subgraph: Subgraph,
7746 body_subgraph: Subgraph,
7765 cond_fn: Subgraph,
7766 body_fn: Subgraph,
|
| H A D | lowering.py | 6250 def associative_scan(combine_fn: ir.Subgraph, input, dim: int):
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | delegate_data.cc | 154 tensorflow::Status RegisterFunctionDefForSubgraphs( in RegisterFunctionDefForSubgraphs()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/kernel/ |
| H A D | flex_attention.py | 92 subgraph: Subgraph,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | subgraph_test.cc | 107 string Subgraph(const string& fed_str, const string& fetch_str, in Subgraph() function in tensorflow::__anon82913e730111::SubgraphTest
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/android_fuzzing/ |
| D | Model.proto | 210 message Subgraph { message
|
| /aosp_15_r20/packages/modules/NeuralNetworks/driver/sample_hidl/ |
| D | SampleDriverFloatXNNPACK.cpp | 154 class Subgraph { class
1554 Subgraph(xnn_runtime_t runtime, std::unordered_set<uint32_t>&& externals, in Subgraph() function in android::nn::sample_driver::Subgraph
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_subgraphs_pass.cc | 170 class Subgraph { class in tensorflow::__anond9260d780111::Encapsulator
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | xnnpack_delegate.cc | 252 class Subgraph { class
4738 Subgraph(const Delegate& delegate, xnn_runtime_t runtime, in Subgraph() function in tflite::xnnpack::__anon4a1244080111::Subgraph
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/core/ |
| H A D | subgraph.cc | 231 Subgraph::Subgraph(ErrorReporter* error_reporter, in Subgraph() function in tflite::Subgraph
|