Searched refs:data_mapper (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 82 void operator()(Scalar* block, const DataMapper& data_mapper, IndexType rows,
88 LinearMapper lm = data_mapper.getLinearMapper(0, col);
269 Scalar* lhsBlock, const typename LhsMapper::SubMapper& data_mapper,
272 LhsPacker()(lhsBlock, data_mapper, rows, depth);
274 EigenLhsPacker()(lhsBlock, data_mapper, depth, rows, /*stride*/ 0,
280 Scalar* rhsBlock, const typename RhsMapper::SubMapper& data_mapper,
283 RhsPacker()(rhsBlock, data_mapper, depth, cols);
285 EigenRhsPacker()(rhsBlock, data_mapper, depth, cols);
|
| D | eigen_mkldnn_contraction_kernel_test.cc | 70 DataMapper data_mapper(src.data(), dims[0]); in TEST() local
73 data_mapper.getSubMapper(slice_start[0], slice_start[1]), in TEST()
|