Searched refs:src_options (Results 1 – 4 of 4) sorted by relevance
29 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()
23 at::TensorOptions src_options; member
30 auto src_options = c10::TensorOptions().device(src.device()).dtype(at::kByte); in storage_copy() local31 auto src_t = at::empty({0}, src_options).set_(src); in storage_copy()
206 grad_fn->src_options = src.options(); in copy_()