| /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
278 if is_consistent(arg_type.__args__[1], module_instance.num_features) and \
279 is_consistent(n.type.__args__[1], module_instance.num_features) and \
292 def calculate_out_dimension(d_in, module_instance, index): argument
296 padding = (module_instance.padding, module_instance.padding) \
297 if isinstance(module_instance.padding, int) else module_instance.padding
298 kernel_size = (module_instance.kernel_size, module_instance.kernel_size) \
299 if isinstance(module_instance.kernel_size, int) else module_instance.kernel_size
300 stride = (module_instance.stride, module_instance.stride) \
301 if isinstance(module_instance.stride, int) else module_instance.stride
[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
301 return gen_embedding_rules(n, symbols, module_instance.embedding_dim, counter)
934 def layer_norm_inference_rule(n: Node, module_instance, symbols, constraints, counter): argument
940 return gen_layer_norm_constraints(n, module_instance.normalized_shape, symbols, counter)
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
986 …return linear_constraints(n, module_instance.in_features, module_instance.out_features, symbols, c…
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
1133 …c2 = BinConstraintT(avg_pool, TensorType([d1, d2, module_instance.output_size[0], module_instance.…
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | scope.h | 181 std::optional<ModuleInstanceInfo> module_instance() const;
198 return (module_instance().has_value() ==
199 rhs.module_instance().has_value()) &&
200 (module_instance().has_value() &&
201 module_instance().value() == rhs.module_instance().value()) &&
|
| 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/test/cpp/jit/ |
| H A D | test_cs_debug_info_serialization.cpp | 51 auto rhs_module = csptr1->module_instance(); in validate_debug_info()
52 auto lhs_module = csptr2->module_instance(); in validate_debug_info()
|
| /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
231 if (module_instance->node()->kind() == prim::GetAttr && in findObserverName()
232 module_instance->node()->s(attr::name).find("_observer_") != in findObserverName()
234 return module_instance->node()->s(attr::name); in findObserverName()
1120 auto module_instance = n->inputs()[0]; in getInvokedMethods() local
1124 if (module_instance == graph->inputs()[0]) { in getInvokedMethods()
1127 module_instance->node()->kind() == prim::GetAttr && in getInvokedMethods()
1128 module_instance->node()->s(attr::name).find("_observer_") == in getInvokedMethods()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_metal.py | 19 module_instance = self
20 scripted_model = torch.jit.script(module_instance)
|
| H A D | test_vulkan.py | 22 module_instance = self
23 scripted_model = torch.jit.script(module_instance)
|
| H A D | test_xnnpack_integration.py | 775 module_instance = self
777 scripted_model = torch.jit.script(module_instance)
779 scripted_model = torch.jit.trace(module_instance, input_data)
1259 module_instance = self
1261 scripted_model = torch.jit.script(module_instance)
1263 scripted_model = torch.jit.trace(module_instance, input_data)
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/ |
| H A D | debug_info.cpp | 39 const auto& opt_module_instance_info = callstack_ptr->module_instance(); in getStackTraceWithModuleHierarchy()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | callstack_debug_info_serialization.cpp | 34 serialize_module_instance_info(cs_ptr->module_instance())); in serialize()
|
| /aosp_15_r20/external/python/apitools/apitools/base/protorpclite/ |
| D | messages_test.py | 1934 module_instance = self.DefineModule(module)
1948 setattr(module_instance, name, message_class)
|