Searched refs:DoMatMul (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 954 template <typename TL2, typename TR2> class DoMatMul>
1059 DoMatMul<TR, TL>::Compute(&this->cache_tr_, left->matrix<TR>(), in Compute()
1064 DoMatMul<TL, TR>::Compute(&this->cache_nt_, left->matrix<TL>(), in Compute()
1078 typename DoMatMul<TL, TR>::TensorInfoCache cache_nt_;
1080 typename DoMatMul<TR, TL>::TensorInfoCache cache_tr_;
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 390 bool DoMatMul(Stream* stream, const DeviceMemory<float>& input_data,
|
| D | cuda_dnn.cc | 3525 bool CudnnSupport::DoMatMul(Stream* stream, in DoMatMul() function in stream_executor::gpu::CudnnSupport
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 381 bool DoMatMul(Stream* stream, const DeviceMemory<float>& input_data,
|
| D | rocm_dnn.cc | 3285 bool MIOpenSupport::DoMatMul(Stream* stream, in DoMatMul() function in stream_executor::gpu::MIOpenSupport
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1498 virtual bool DoMatMul(Stream* stream, const DeviceMemory<float>& input_data,
|
| D | stream.cc | 1149 CheckError(dnn->DoMatMul(this, input_data, weights, input_dimensions, in ThenMatMul()
|