Searched refs:output_leaf_index_tensor (Results 1 – 1 of 1) sorted by relevance
113 Tensor output_leaf_index_tensor(DT_INT32, TensorShape({2, 2})); in TEST_F() local116 &output_leaf_index_tensor); in TEST_F()119 EXPECT_FLOAT_EQ(0, output_leaf_index_tensor.matrix<int>()( in TEST_F()121 EXPECT_FLOAT_EQ(0, output_leaf_index_tensor.matrix<int>()( in TEST_F()123 EXPECT_FLOAT_EQ(2, output_leaf_index_tensor.matrix<int>()( in TEST_F()125 EXPECT_FLOAT_EQ(1, output_leaf_index_tensor.matrix<int>()( in TEST_F()