Searched defs:named_buffers (Results 1 – 11 of 11) sorted by relevance
352 def named_buffers( member in NamedMemberAccessor
909 def functional_call(named_params, named_buffers, *args, **kwargs): argument
162 OrderedDict<std::string, Tensor> Module::named_buffers(bool recurse) const { in named_buffers() function in torch::nn::Module
295 auto named_buffers = module->named_buffers(); in TEST_F() local304 auto named_buffers = module->named_buffers(); in TEST_F() local
748 def named_buffers(self) -> Iterator[Tuple[str, torch.Tensor]]: member in ExportedProgram
522 named_buffer_list Module::named_buffers(bool recurse) const { in named_buffers() function in torch::jit::Module
418 def named_buffers( # type: ignore[return] member in _RemoteModule
981 def named_buffers( member in FullyShardedDataParallel
2691 def named_buffers( member in Module
9512 def buffer_comm_hook(ddp, named_buffers): argument9612 def buffer_comm_hook(ddp, named_buffers): argument
336 def names(named_buffers): argument