Searched defs:opt_tensor (Results 1 – 3 of 3) sorted by relevance
432 def consume_optional(opt_tensor): argument
25 auto opt_tensor = (i % 2 == 0) ? std::optional<at::Tensor>(at::empty({0})) : std::nullopt; in get_boxed_opt_tensor_vector() local
814 for (const auto& opt_tensor : tensors_to_save) { in _save_variables() local