Home
last modified time | relevance | path

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

/external/executorch/extension/training/optimizer/
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()
Dsgd.cpp70 SGDParamGroup::named_parameters() const { in named_parameters() function in executorch::extension::training::optimizer::SGDParamGroup
/external/executorch/extension/training/optimizer/test/
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
/external/executorch/extension/training/module/
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
/external/executorch/extension/training/pybindings/
D_training_module.py67 def named_parameters(self) -> Dict[str, Tensor]: member in TrainingModule
/external/tensorflow/tensorflow/python/kernel_tests/proto/
Dproto_op_test_base.py37 def named_parameters(extension=True): member in ProtoOpTestBase
/external/pytorch/torch/csrc/api/src/nn/modules/
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
/external/cronet/stable/third_party/protobuf/python/google/protobuf/internal/
D_parameterized.py325 def named_parameters(*testcases): # pylint: disable=invalid-name function
/external/cronet/tot/third_party/protobuf/python/google/protobuf/internal/
D_parameterized.py325 def named_parameters(*testcases): # pylint: disable=invalid-name function
/external/protobuf/python/google/protobuf/internal/
D_parameterized.py325 def named_parameters(*testcases): # pylint: disable=invalid-name function
/external/pytorch/torch/distributed/algorithms/_checkpoint/
Dcheckpoint_wrapper.py56 def named_parameters( member in ActivationWrapper
/external/pytorch/torch/nn/utils/
D_named_member_accessor.py345 def named_parameters( member in NamedMemberAccessor
/external/python/absl-py/absl/testing/
Dparameterized.py465 def named_parameters(*testcases): function
/external/pytorch/torch/csrc/jit/mobile/
Dmodule.cpp179 const std::map<std::string, at::Tensor> Module::named_parameters() const { in named_parameters() function in torch::jit::mobile::Module
/external/pytorch/torch/csrc/api/src/nn/
Dmodule.cpp139 OrderedDict<std::string, Tensor> Module::named_parameters(bool recurse) const { in named_parameters() function in torch::nn::Module
/external/pytorch/torch/export/
Dexported_program.py731 def named_parameters(self) -> Iterator[Tuple[str, torch.nn.Parameter]]: member in ExportedProgram
/external/pytorch/torch/csrc/jit/api/
Dmodule.cpp542 named_parameter_list Module::named_parameters(bool recurse) const { in named_parameters() function in torch::jit::Module
/external/pytorch/torch/distributed/nn/api/
Dremote_module.py410 def named_parameters( # type: ignore[return] member in _RemoteModule
/external/pytorch/torch/distributed/fsdp/
Dfully_sharded_data_parallel.py999 def named_parameters( member in FullyShardedDataParallel
/external/pytorch/torch/nn/modules/
Dmodule.py2636 def named_parameters( member in Module
/external/pytorch/test/
Dtest_nn.py282 def names(named_parameters): argument
308 def names(named_parameters): argument
/external/pytorch/test/cpp/api/
Dmodules.cpp3575 auto named_parameters = multihead_attn_module->named_parameters(); in _multihead_attn_test_helper() local