Home
last modified time | relevance | path

Searched defs:compute_type (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op.cc219 se::blas::ComputationType* compute_type) { in GetCublasAutotuneComputationType()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1009 cudnnDataType_t compute_type, in CudnnRnnDescriptor()
1036 cudnnDataType_t data_type, cudnnDataType_t compute_type, in Create()
1116 cudnnDataType_t compute_type() const { return compute_type_; } in compute_type() function in stream_executor::gpu::CudnnRnnDescriptor
2809 cudnnDataType_t compute_type; in DoConvolve() local
Dcuda_blas.cc2220 cudaDataType_t compute_type = in DoBlasGemmBatchedInternal() local