Home
last modified time | relevance | path

Searched defs:get_input (Results 1 – 25 of 69) sorted by relevance

123

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor_strided_blocks.py160 def get_input() -> torch.Tensor: function
208 def get_input(view_size: Tuple[int]) -> torch.Tensor: function
300 def get_input() -> torch.Tensor: function
421 def get_input() -> torch.Tensor: function
/aosp_15_r20/external/webrtc/rtc_tools/py_event_log_analyzer/
H A Dmisc.py83 get_input = raw_input # pylint: disable=invalid-name variable
85 get_input = input # pylint: disable=invalid-name variable
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/spirit/include/boost/spirit/home/support/iterators/detail/
Dcombine_policies.hpp108 inline static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function
196 inline static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function
297 inline static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function
407 inline static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function
Dbuffering_input_iterator_policy.hpp65 static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function in boost::spirit::iterator_policies::buffering_input_iterator::unique
111 result_type& get_input() in get_input() function
Dsplit_functor_input_policy.hpp79 static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function in boost::spirit::iterator_policies::split_functor_input::unique
142 static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function in boost::spirit::iterator_policies::split_functor_input::unique
Dlex_input_policy.hpp41 static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function in boost::spirit::iterator_policies::lex_input::unique
Dinput_iterator_policy.hpp70 static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function in boost::spirit::iterator_policies::input_iterator::unique
Dfunctor_input_policy.hpp62 static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function in boost::spirit::iterator_policies::functor_input::unique
Distream_policy.hpp48 static typename MultiPass::reference get_input(MultiPass& mp) in get_input() function in boost::spirit::iterator_policies::istream::unique
/aosp_15_r20/external/pytorch/test/distributed/checkpoint/
H A Dtest_hsdp_checkpoint.py45 def get_input(self): member in SimpleModel
65 def get_input(self): member in SimpleModelUneven
H A Dtest_fsdp_optim_state.py44 def get_input(self): member in FsdpOptimStateCheckpoint._create_model.TestDummyModel
H A Dtest_format_utils.py42 def get_input(self): member in SimpleModelUneven
/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_dtensor_state_dict.py43 def get_input(self): member in TestDummyModel
59 def get_input(self): member in TestDummyModelUneven
H A Dtest_fsdp_flatten_params.py65 def get_input(device, dtype): function
185 def get_input(self, device, dtype): member in TestFlattenParams._get_empty_module.EmptyModule
H A Dtest_fsdp_use_orig_params.py778 def get_input(self, device: torch.device) -> Tuple[torch.Tensor, ...]: member in TestFSDPUseOrigParamsParamAccess._test_access_params_after_forward.Model
872 def get_input(self, device: torch.device) -> Tuple[torch.Tensor, ...]: member in TestFSDPUseOrigParamsWriteback.Model
H A Dtest_hsdp_dtensor_state_dict.py45 def get_input(self): member in DenseModel
H A Dtest_fsdp_exec_order.py66 def get_input(self, device: torch.device): member in Model
H A Dtest_fsdp_optim_state.py164 def get_input(self, device): member in NestedModel
306 def get_input(self): member in TestDummyModel
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_fsdp.py90 def get_input(self, device) -> Tuple[torch.Tensor, ...]: member in FSDPTestModel
279 def get_input(self, device): member in TransformerWithSharedParams
417 def get_input(self, device): member in NestedWrappedModule
623 def get_input(self, device): member in ModuleWithDelay
/aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/
H A Dtest_2d_composability.py73 def get_input(self): member in SimpleModel
94 def get_input(self): member in SimpleModelUneven
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dpre_grad.py548 def get_input(self) -> torch.fx.Node: member in NormalizedLinearNode
573 def get_input(self) -> torch.fx.Node: member in NormalizedMatmulNode
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/spirit/include/boost/spirit/home/classic/iterator/
Dmulti_pass.hpp568 reference get_input() const in get_input() function in boost::spirit::multi_pass_policies::input_iterator::inner
641 reference get_input() const in get_input() function in boost::spirit::multi_pass_policies::lex_input::inner
722 reference get_input() const in get_input() function in boost::spirit::multi_pass_policies::functor_input::inner
/aosp_15_r20/external/pytorch/benchmarks/transformer/
H A Dsdpa.py69 def get_input( function
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/examples/
H A Dstateful_example.py38 def get_input(self): member in Model
/aosp_15_r20/external/pytorch/test/quantization/core/experimental/
H A Dtest_float8.py216 def test_cast_round_trip(self, dtype, get_input, device): argument

123