Searched refs:lhs_block (Results 1 – 3 of 3) sorted by relevance
| /external/gemmlowp/internal/ |
| D | multi_thread_gemm.h | 673 auto lhs_block = lhs.block(start_row, 0, block_rows, depth); in MultiThreadGemm() local
680 new TaskType(context, kernel, lhs_block, packed_rhs, result, in MultiThreadGemm()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deep_conv2d.cc | 508 T* lhs_block) in GemmFilterPacker() argument
511 lhs_block_(lhs_block), in GemmFilterPacker()
589 const int64 out_buffer_size, const T* lhs_block, const T* rhs_input, in GemmState() argument
595 lhs_block_(lhs_block), in GemmState()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContractionCuda.h | 322 const Scalar* lhs_block = &lhs_shmem[threadIdx.x + 9 * threadIdx.y]; in EigenContractionKernelInternal() local
325 #define lhs_element(i, j) lhs_block[72 * ((i) + 8 * (j))] in EigenContractionKernelInternal()
|