Searched refs:algorithm_idx (Results 1 – 4 of 4) sorted by relevance
31 ThunkInfo thunk_info, cublas_lt::MatmulPlan plan, int64_t algorithm_idx, in CublasLtMatmulThunk() argument37 algorithm_idx_(algorithm_idx), in CublasLtMatmulThunk()
34 int64_t algorithm_idx, BufferAllocation::Slice a_buffer,
486 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm();488 0 <= algorithm_idx && algorithm_idx < algorithms.size(),490 const auto& algorithm = algorithms[algorithm_idx];
314 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm(); in operator ()() local315 algorithm = algorithms[algorithm_idx]; in operator ()()