Searched refs:lhs_block (Results 1 – 5 of 5) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContraction.h | 86 LhsScalar** lhs_block,
88 eigen_assert(lhs_block);
93 *lhs_block = reinterpret_cast<LhsScalar*>(block_mem);
219 EIGEN_DEVICE_FUNC BlockMemHandle allocate(Device& d, LhsBlock* lhs_block,
221 return BlockMemAllocator::allocate(d, bm, bk, bn, lhs_block, rhs_block);
|
| D | TensorContractionGpu.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()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 591 EIGEN_DEVICE_FUNC BlockMemHandle allocate(Device& d, LhsBlock* lhs_block, \
594 d, bm, bk, bn, &lhs_block->packed_data, &rhs_block->packed_data); \
786 EIGEN_DEVICE_FUNC BlockMemHandle allocate(Device& d, LhsBlock* lhs_block, \
789 d, bm, bk, bn, &lhs_block->packed_data, &rhs_block->packed_data); \
|
| D | deep_conv2d.cc | 517 T* lhs_block) in GemmFilterPacker() argument
520 lhs_block_(lhs_block), in GemmFilterPacker()
601 const int64_t out_buffer_size, const T* lhs_block, in GemmState() argument
607 lhs_block_(lhs_block), in GemmState()
|
| /external/gemmlowp/internal/ |
| D | multi_thread_gemm.h | 701 auto lhs_block = lhs.block(start_row, 0, block_rows, depth); in MultiThreadGemm() local
708 new TaskType(context, kernel, lhs_block, packed_rhs, result, in MultiThreadGemm()
|