Home
last modified time | relevance | path

Searched refs:src_options (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/csrc/autograd/functions/
Dtensor.cpp29 const auto src_type = src_options.dtype().toScalarType(); in apply()
34 at::DeviceGuard device_guard(src_options.device()); in apply()
35 grad_inputs[1] = grad->to(src_options); in apply()
42 args.collect(src_options); in compiled_args()
47 saved.before(src_options); in apply_with_saved()
49 saved.after(src_options); in apply_with_saved()
Dtensor.h23 at::TensorOptions src_options; member
/external/pytorch/aten/src/ATen/
DStorageUtils.cpp30 auto src_options = c10::TensorOptions().device(src.device()).dtype(at::kByte); in storage_copy() local
31 auto src_t = at::empty({0}, src_options).set_(src); in storage_copy()
/external/pytorch/torch/csrc/autograd/
DVariableTypeManual.cpp206 grad_fn->src_options = src.options(); in copy_()