Searched refs:Tensor2d (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_mkldnn_contraction_kernel_test.cc | 47 using Tensor2d = Tensor<Scalar, 2, Options, Index>; in TEST() typedef
52 Tensor2d src(dims); in TEST()
66 Tensor2d pack_dst(slice_size[0], slice_size[1]); in TEST()
67 Tensor2d slice_dst(slice_size[0], slice_size[1]); in TEST()
94 using Tensor2d = Tensor<Scalar, 2, Options, Index>; in TEST() typedef
100 Tensor2d lhs(m, k); in TEST()
103 Tensor2d rhs(k, n); in TEST()
111 Tensor2d mkldnn_result(m, n); in TEST()
|