Home
last modified time | relevance | path

Searched defs:named_parameters (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/executorch/extension/training/optimizer/
H A Dsgd.h154 const std::map<executorch::aten::string_view, executorch::aten::Tensor>& in SGDParamGroup()
158 const std::map<executorch::aten::string_view, executorch::aten::Tensor>& in SGDParamGroup()
192 const std::map<executorch::aten::string_view, executorch::aten::Tensor>& in SGD()
H A Dsgd.cpp70 SGDParamGroup::named_parameters() const { in named_parameters() function in executorch::extension::training::optimizer::SGDParamGroup
/aosp_15_r20/external/executorch/extension/training/optimizer/test/
H A Dsgd_test.cpp71 std::map<exec_aten::string_view, exec_aten::Tensor> named_parameters; in TEST_F() local
93 std::map<exec_aten::string_view, exec_aten::Tensor> named_parameters; in TEST_F() local
/aosp_15_r20/external/executorch/extension/training/module/
H A Dtraining_module.cpp88 TrainingModule::named_parameters(const std::string& method_name) { in named_parameters() function in executorch::extension::training::TrainingModule
89 std::map<exec_aten::string_view, exec_aten::Tensor> named_parameters; in named_parameters() local
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Drnn.cpp167 auto named_parameters = this->named_parameters(/*recurse=*/false); in reset() local
246 auto named_parameters = this->named_parameters(/*recurse=*/false); in reset_flat_weights() local
385 auto named_parameters = this->named_parameters(/*recurse=*/false); in all_weights() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/proto/
H A Dproto_op_test_base.py37 def named_parameters(extension=True): member in ProtoOpTestBase
/aosp_15_r20/external/executorch/extension/training/pybindings/
H A D_training_module.py67 def named_parameters(self) -> Dict[str, Tensor]: member in TrainingModule
/aosp_15_r20/frameworks/libs/binary_translation/guest_abi/arm/include/berberis/guest_abi/
Dguest_params_arch.h250 GuestVAListParams(const GuestParamsValues<Func, kCallingConventionsVariant>&& named_parameters) in GuestVAListParams()
/aosp_15_r20/frameworks/libs/binary_translation/guest_abi/riscv64/include/berberis/guest_abi/
Dguest_params_arch.h234 GuestVAListParams(const GuestParamsValues<Func, kCallingConventionsVariant>&& named_parameters) in GuestVAListParams()
/aosp_15_r20/external/cronet/third_party/protobuf/python/google/protobuf/internal/
H A D_parameterized.py325 def named_parameters(*testcases): # pylint: disable=invalid-name function
/aosp_15_r20/external/protobuf/python/google/protobuf/internal/
H A D_parameterized.py325 def named_parameters(*testcases): # pylint: disable=invalid-name function
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_checkpoint/
H A Dcheckpoint_wrapper.py56 def named_parameters( member in ActivationWrapper
/aosp_15_r20/frameworks/libs/binary_translation/guest_abi/arm64/include/berberis/guest_abi/
Dguest_params_arch.h208 GuestVAListParams(const GuestParamsValues<Func, kCallingConventionsVariant>&& named_parameters) in GuestVAListParams()
/aosp_15_r20/external/python/absl-py/absl/testing/
Dparameterized.py465 def named_parameters(*testcases): function
/aosp_15_r20/external/pytorch/torch/nn/utils/
H A D_named_member_accessor.py345 def named_parameters( member in NamedMemberAccessor
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/
H A Dmodule.cpp179 const std::map<std::string, at::Tensor> Module::named_parameters() const { in named_parameters() function in torch::jit::mobile::Module
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/
H A Dmodule.cpp139 OrderedDict<std::string, Tensor> Module::named_parameters(bool recurse) const { in named_parameters() function in torch::nn::Module
/aosp_15_r20/external/pytorch/torch/export/
H A Dexported_program.py731 def named_parameters(self) -> Iterator[Tuple[str, torch.nn.Parameter]]: member in ExportedProgram
/aosp_15_r20/external/pytorch/torch/csrc/jit/api/
H A Dmodule.cpp542 named_parameter_list Module::named_parameters(bool recurse) const { in named_parameters() function in torch::jit::Module
/aosp_15_r20/external/pytorch/torch/distributed/nn/api/
H A Dremote_module.py410 def named_parameters( # type: ignore[return] member in _RemoteModule
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A Dfully_sharded_data_parallel.py999 def named_parameters( member in FullyShardedDataParallel
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dmodule.py2636 def named_parameters( member in Module
/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py282 def names(named_parameters): argument
308 def names(named_parameters): argument
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dmodules.cpp3575 auto named_parameters = multihead_attn_module->named_parameters(); in _multihead_attn_test_helper() local