/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)
|