Searched refs:MatMulTypes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op.h | 33 struct MatMulTypes { struct
53 const Device& d, typename MatMulTypes<T>::out_type out, argument
54 typename MatMulTypes<T>::in_type in0,
55 typename MatMulTypes<T>::in_type in1,
|
| D | matmul_op.cc | 536 const CPUDevice& d, typename MatMulTypes<T>::out_type out, in operator ()()
537 typename MatMulTypes<T>::in_type in0, in operator ()()
538 typename MatMulTypes<T>::in_type in1, in operator ()()
549 const SYCLDevice& d, typename MatMulTypes<T>::out_type out, in operator ()()
550 typename MatMulTypes<T>::in_type in0, in operator ()()
551 typename MatMulTypes<T>::in_type in1, in operator ()()
|