Home
last modified time | relevance | path

Searched refs:pack_dst (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc66 Tensor2d pack_dst(slice_size[0], slice_size[1]); in TEST() local
72 gemm_pack(pack_dst.data(), in TEST()
80 EXPECT_EQ(pack_dst.dimensions().TotalSize(), in TEST()
82 for (size_t i = 0; i < pack_dst.dimensions().TotalSize(); ++i) { in TEST()
83 Scalar packed = pack_dst.coeff(i); in TEST()