Searched refs:rhs_block (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 592 RhsBlock* rhs_block) { \
594 d, bm, bk, bn, &lhs_block->packed_data, &rhs_block->packed_data); \
787 RhsBlock* rhs_block) { \
789 d, bm, bk, bn, &lhs_block->packed_data, &rhs_block->packed_data); \
|
| D | deep_conv2d.cc | 590 T* rhs_block, T* out_buffer) in GemmState() argument
596 rhs_block_(rhs_block), in GemmState()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContractionCuda.h | 323 const Scalar* rhs_block = &rhs_shmem[threadIdx.x + 8 * threadIdx.z]; in EigenContractionKernelInternal() local
326 #define rhs_element(i, j) rhs_block[72 * ((i) + 8 * (j))] in EigenContractionKernelInternal()
|