Searched refs:SingleThreadedMatMulDispatch (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 67 void SingleThreadedMatMulDispatch(const void* run_options_ptr, T* out, T* lhs, in SingleThreadedMatMulDispatch() function
92 SingleThreadedMatMulDispatch<Eigen::half>(run_options_ptr, out, lhs, rhs, m, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
101 SingleThreadedMatMulDispatch<float>(run_options_ptr, out, lhs, rhs, m, n, k, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
110 SingleThreadedMatMulDispatch<double>(run_options_ptr, out, lhs, rhs, m, n, k, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
120 SingleThreadedMatMulDispatch<std::complex<float>>( in __xla_cpu_runtime_EigenSingleThreadedMatMulC64()
130 SingleThreadedMatMulDispatch<std::complex<double>>( in __xla_cpu_runtime_EigenSingleThreadedMatMulC128()
140 SingleThreadedMatMulDispatch<tensorflow::int32>( in __xla_cpu_runtime_EigenSingleThreadedMatMulS32()
|