Searched refs:Tensor2d (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_mkldnn_contraction_kernel_test.cc | 54 using Tensor2d = Tensor<Scalar, 2, Options, Index>; in TEST() typedef
59 Tensor2d src(dims); in TEST()
73 Tensor2d pack_dst(slice_size[0], slice_size[1]); in TEST()
74 Tensor2d slice_dst(slice_size[0], slice_size[1]); in TEST()
101 using Tensor2d = Tensor<Scalar, 2, Options, Index>; in TEST() typedef
107 Tensor2d lhs(m, k); in TEST()
110 Tensor2d rhs(k, n); in TEST()
118 Tensor2d mkldnn_result(m, n); in TEST()
|