Searched refs:to_impl (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/csrc/api/src/nn/ |
| D | module.cpp | 239 to_impl(device, dtype, non_blocking); in to()
243 to_impl(dtype, non_blocking); in to()
247 to_impl(device, non_blocking); in to()
|
| /external/pytorch/torch/csrc/jit/api/ |
| D | module.cpp | 157 to_impl(device, dtype, non_blocking); in to()
161 to_impl(/*device=*/std::nullopt, dtype, non_blocking); in to()
165 to_impl(device, /*dtype=*/std::nullopt, non_blocking); in to()
182 void Module::to_impl( in to_impl() function in torch::jit::Module
|
| D | module.h | 319 void to_impl(
|
| /external/pytorch/torch/csrc/api/include/torch/nn/ |
| D | module.h | 584 void to_impl(Ts&&... ts);
686 void Module::to_impl(Ts&&... ts) { in to_impl() function
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorConversions.cpp | 417 static inline Tensor to_impl( in to_impl() function
451 return to_impl( in _autocast_to_reduced_precision()
465 return to_impl( in _autocast_to_full_precision()
482 return to_impl( in to()
494 return to_impl( in to()
506 return to_impl( in to()
519 return to_impl( in to()
|