Searched defs:workspace_ptr (Results 1 – 5 of 5) sorted by relevance
367 auto workspace_ptr = in run_conv_plan() local414 auto workspace_ptr = in run_conv_plan_fused() local650 at::DataPtr& workspace_ptr) { in generate_and_filter_plans()720 at::DataPtr workspace_ptr; in get_plans_from_find() local775 at::DataPtr workspace_ptr; in get_plans_from_find_fused() local
602 auto workspace_ptr = in run_cudnn_SDP_fprop() local695 auto workspace_ptr = in run_cudnn_SDP_bprop() local
119 auto workspace_ptr = c10::cuda::CUDACachingAllocator::get()->allocate(workspace_size); in apply_impl_helper() local
147 auto workspace_ptr = c10::cuda::CUDACachingAllocator::get()->allocate(workspace_size); in apply_impl_helper() local
4199 uint8* workspace_ptr = 0; in DoPoolBackward() local