/aosp_15_r20/external/executorch/extension/training/optimizer/ |
H A D | sgd.h | 154 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 D | sgd.cpp | 70 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 D | sgd_test.cpp | 71 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 D | training_module.cpp | 88 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 D | rnn.cpp | 167 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 D | proto_op_test_base.py | 37 def named_parameters(extension=True): member in ProtoOpTestBase
|
/aosp_15_r20/external/executorch/extension/training/pybindings/ |
H A D | _training_module.py | 67 def named_parameters(self) -> Dict[str, Tensor]: member in TrainingModule
|
/aosp_15_r20/frameworks/libs/binary_translation/guest_abi/arm/include/berberis/guest_abi/ |
D | guest_params_arch.h | 250 GuestVAListParams(const GuestParamsValues<Func, kCallingConventionsVariant>&& named_parameters) in GuestVAListParams()
|
/aosp_15_r20/frameworks/libs/binary_translation/guest_abi/riscv64/include/berberis/guest_abi/ |
D | guest_params_arch.h | 234 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.py | 325 def named_parameters(*testcases): # pylint: disable=invalid-name function
|
/aosp_15_r20/external/protobuf/python/google/protobuf/internal/ |
H A D | _parameterized.py | 325 def named_parameters(*testcases): # pylint: disable=invalid-name function
|
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_checkpoint/ |
H A D | checkpoint_wrapper.py | 56 def named_parameters( member in ActivationWrapper
|
/aosp_15_r20/frameworks/libs/binary_translation/guest_abi/arm64/include/berberis/guest_abi/ |
D | guest_params_arch.h | 208 GuestVAListParams(const GuestParamsValues<Func, kCallingConventionsVariant>&& named_parameters) in GuestVAListParams()
|
/aosp_15_r20/external/python/absl-py/absl/testing/ |
D | parameterized.py | 465 def named_parameters(*testcases): function
|
/aosp_15_r20/external/pytorch/torch/nn/utils/ |
H A D | _named_member_accessor.py | 345 def named_parameters( member in NamedMemberAccessor
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/ |
H A D | module.cpp | 179 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 D | module.cpp | 139 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 D | exported_program.py | 731 def named_parameters(self) -> Iterator[Tuple[str, torch.nn.Parameter]]: member in ExportedProgram
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/api/ |
H A D | module.cpp | 542 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 D | remote_module.py | 410 def named_parameters( # type: ignore[return] member in _RemoteModule
|
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
H A D | fully_sharded_data_parallel.py | 999 def named_parameters( member in FullyShardedDataParallel
|
/aosp_15_r20/external/pytorch/torch/nn/modules/ |
H A D | module.py | 2636 def named_parameters( member in Module
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_nn.py | 282 def names(named_parameters): argument 308 def names(named_parameters): argument
|
/aosp_15_r20/external/pytorch/test/cpp/api/ |
H A D | modules.cpp | 3575 auto named_parameters = multihead_attn_module->named_parameters(); in _multihead_attn_test_helper() local
|