Searched refs:LaunchMatMul (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op.cc | 42 struct LaunchMatMul;
147 struct LaunchMatMul<CPUDevice, T, USE_CUBLAS> : public LaunchMatMulCPU<T> {}; struct
154 struct LaunchMatMul<SYCLDevice, T, USE_CUBLAS> : public LaunchMatMulSYCL<T> {}; struct
252 struct LaunchMatMul<GPUDevice, T, true /* USE_CUBLAS */> { struct
446 LaunchMatMul<Device, T, USE_CUBLAS>::GetBlasGemmAlgorithm( in MatMulOp()
511 LaunchMatMul<Device, float, USE_CUBLAS>::launch( in Compute()
517 LaunchMatMul<Device, T, USE_CUBLAS>::launch( in Compute()
|