Searched defs:module_instance (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | graph_gradual_typechecker.py | 260 def bn2d_inference_rule(n: Node, module_instance): argument
292 def calculate_out_dimension(d_in, module_instance, index): argument
337 def conv2d_inference_rule(n: Node, module_instance): argument
365 def relu_inference_rule(n: Node, module_instance): argument
379 def maxpool2d_check(typ, module_instance): argument
401 def maxpool2d_inference_rule(n: Node, module_instance): argument
422 def linear_check(tensor_type, module_instance): argument
439 def linear_inference_rule(n: Node, module_instance): argument
453 def adaptiveavgpool2d_check(tensor_type, module_instance): argument
476 def adaptiveavgpool2d_inference_rule(n: Node, module_instance): argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/migrate_gradual_types/ |
| H A D | constraint_generator.py | 296 def embedding_inference_rule(n: Node, module_instance, symbols, constraints, counter): argument
934 def layer_norm_inference_rule(n: Node, module_instance, symbols, constraints, counter): argument
967 def relu_inference_rule(n: Node, module_instance, symbols, constraints, counter): argument
980 def linear_inference_rule(n: Node, module_instance, symbols, constraints, counter): argument
1096 def batchnorm_inference_rule(n: Node, module_instance, symbols, constraints, counter): argument
1118 def adaptive_inference_rule(n: Node, module_instance, symbols, constraints, counter): argument
1139 def conv2d_inference_rule(n: Node, module_instance, symbols, constraints, counter): argument
1168 def maxpool_inference_rule(n: Node, module_instance, symbols, constraints, counter): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | scope.cpp | 172 std::optional<ModuleInstanceInfo> InlinedCallStack::module_instance() const { in module_instance() function in torch::jit::InlinedCallStack
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | insert_quant_dequant.cpp | 230 auto module_instance = n->inputs().at(0); in findObserverName() local
1120 auto module_instance = n->inputs()[0]; in getInvokedMethods() local
|