Searched refs:data_mapper (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 89 void operator()(Scalar* block, const DataMapper& data_mapper, IndexType rows,
95 LinearMapper lm = data_mapper.getLinearMapper(0, col);
400 static bool block(const typename DataMapper::SubMapper& data_mapper,
433 static bool block(const typename DataMapper::SubMapper& data_mapper, \
440 const StorageIndex vert_offset = data_mapper.vert_offset(); \
441 const StorageIndex horiz_offset = data_mapper.horiz_offset(); \
443 Side == Lhs ? data_mapper.base_mapper().stride() \
444 : data_mapper.base_mapper().nocontract_strides()[0]; \
445 const Scalar* data = data_mapper.base_mapper().tensor().data(); \
631 LhsBlock* lhsBlock, const typename LhsMapper::SubMapper& data_mapper, \
[all …]
|
| D | eigen_mkldnn_contraction_kernel_test.cc | 77 DataMapper data_mapper(src.data(), dims[0]); in TEST() local
80 data_mapper.getSubMapper(slice_start[0], slice_start[1]), in TEST()
|