Home
last modified time | relevance | path

Searched refs:module_instance (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dgraph_gradual_typechecker.py260 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 Dconstraint_generator.py296 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 Dscope.h181 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 Dscope.cpp172 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 Dtest_cs_debug_info_serialization.cpp51 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 Dinsert_quant_dequant.cpp230 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 Dtest_metal.py19 module_instance = self
20 scripted_model = torch.jit.script(module_instance)
H A Dtest_vulkan.py22 module_instance = self
23 scripted_model = torch.jit.script(module_instance)
H A Dtest_xnnpack_integration.py775 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 Ddebug_info.cpp39 const auto& opt_module_instance_info = callstack_ptr->module_instance(); in getStackTraceWithModuleHierarchy()
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dcallstack_debug_info_serialization.cpp34 serialize_module_instance_info(cs_ptr->module_instance())); in serialize()
/aosp_15_r20/external/python/apitools/apitools/base/protorpclite/
Dmessages_test.py1934 module_instance = self.DefineModule(module)
1948 setattr(module_instance, name, message_class)