Searched refs:set_scratch_size (Results 1 – 5 of 5) sorted by relevance
275 algo->set_scratch_size(selected_result->scratch_bytes()); in BestCudnnConvAlgorithm()
232 algorithm.set_scratch_size(*options.scratch_size_override); in RunGpuConvImpl()
788 void set_scratch_size(size_t val) { scratch_size_ = val; } in set_scratch_size() function831 void set_scratch_size(size_t val) { scratch_size_ = val; } in set_scratch_size() function
510 profile_result.set_scratch_size(solution.workspace_size); in GetProfileResultFromConvSolution()536 profile_result.set_scratch_size(algorithm.memory); in GetProfileResultFromConvAlgoPerf()3069 output_profile_result->set_scratch_size(scratch_memory.size()); in DoConvolve()
3169 output_profile_result->set_scratch_size(scratch_memory.size()); in DoConvolve()3288 output_profile_result->set_scratch_size(scratch.size()); in DoFusedConvolveImpl()