Searched defs:tuning_ctx (Results 1 – 2 of 2) sorted by relevance
720 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in bgemm() local731 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in bgemm() local742 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in bgemm() local753 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in bgemm() local764 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in bgemm() local775 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in bgemm() local1116 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in gemm() local1127 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in gemm() local1138 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in gemm() local1149 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in gemm() local[all …]
383 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in addmm_out_cuda_impl() local1075 auto tuning_ctx = at::cuda::tunable::getTuningContext(); in _scaled_mm_out_cuda() local