Searched refs:kDimSize (Results 1 – 1 of 1) sorted by relevance
25 const int kDimSize = 2; variable28 auto eigen_tensor = tensor->tensor<float, kDimSize>(); in InitTensor()29 for (int i = 0; i < kDimSize; ++i) { in InitTensor()30 for (int j = 0; j < kDimSize; ++j) { in InitTensor()37 auto eigen_tensor1 = tensor1.tensor<float, kDimSize>(); in Equal()38 auto eigen_tensor2 = tensor2.tensor<float, kDimSize>(); in Equal()39 for (int i = 0; i < kDimSize; ++i) { in Equal()40 for (int j = 0; j < kDimSize; ++j) { in Equal()50 Tensor input(DT_FLOAT, TensorShape({kDimSize, kDimSize})); in TEST()51 Tensor output(DT_FLOAT, TensorShape({kDimSize, kDimSize})); in TEST()