Searched refs:SingleThreadedMatMulDispatch (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 70 void SingleThreadedMatMulDispatch(const void* run_options_ptr, T* out, T* lhs, in SingleThreadedMatMulDispatch() function
96 SingleThreadedMatMulDispatch<Eigen::half>(run_options_ptr, out, lhs, rhs, m, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
106 SingleThreadedMatMulDispatch<float>(run_options_ptr, out, lhs, rhs, m, n, k, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
116 SingleThreadedMatMulDispatch<double>(run_options_ptr, out, lhs, rhs, m, n, k, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
|