Searched refs:numel_in_stack (Results 1 – 1 of 1) sorted by relevance
116 int64_t numel_in_stack = first_tensor.numel() * tensors.size(); in can_use_native_serial_stack_impl() local117 return numel_in_stack < at::internal::GRAIN_SIZE || at::get_num_threads() == 1; in can_use_native_serial_stack_impl()