Searched refs:compute_type (Results 1 – 3 of 3) sorted by relevance
219 se::blas::ComputationType* compute_type) { in GetCublasAutotuneComputationType() argument226 *compute_type = ComputationType::kF32; in GetCublasAutotuneComputationType()228 *compute_type = ComputationType::kF16; in GetCublasAutotuneComputationType()232 *compute_type = ComputationType::kF32; in GetCublasAutotuneComputationType()235 *compute_type = ComputationType::kF64; in GetCublasAutotuneComputationType()
1009 cudnnDataType_t compute_type, in CudnnRnnDescriptor() argument1024 compute_type_(compute_type), in CudnnRnnDescriptor()1036 cudnnDataType_t data_type, cudnnDataType_t compute_type, in Create() argument1052 /*dataType=*/compute_type)); in Create()1103 compute_type, algorithm_config, in Create()1116 cudnnDataType_t compute_type() const { return compute_type_; } in compute_type() function in stream_executor::gpu::CudnnRnnDescriptor2809 cudnnDataType_t compute_type; in DoConvolve() local2812 &convolution_mode, &compute_type)); in DoConvolve()
2220 cudaDataType_t compute_type = in DoBlasGemmBatchedInternal() local2234 batch_count, compute_type, algo); in DoBlasGemmBatchedInternal()