Searched refs:output_mapper (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 290 const OutputMapper& output_mapper, const Scalar* lhsBlock,
294 GemmKernel()(output_mapper, lhsBlock, rhsBlock, rows, depth, cols, alpha);
296 GebpKernel()(output_mapper, lhsBlock, rhsBlock, rows, depth, cols, alpha,
|
| D | eigen_mkldnn_contraction_kernel_test.cc | 113 OutputMapper output_mapper(mkldnn_result.data(), m); in TEST() local
116 gemm_kernel(output_mapper, lhs.data(), rhs.data(), m, k, n, /*alpha=*/1.0); in TEST()
|
| D | conv_ops_fused_impl.h | 163 const ContractionOutputMapper<Scalar, Index>& output_mapper, in operator()
172 T* output_base = &output_mapper(0, col); in operator()
196 const ContractionOutputMapper<Scalar, Index>& output_mapper, in operator()
211 T* output_base = &output_mapper(0, col); in operator()
|