Searched refs:lhs_block (Results 1 – 4 of 4) sorted by relevance
| /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 | 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/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()
|
| /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()
|