Searched refs:sparse_float_shape2 (Results 1 – 2 of 2) sorted by relevance
55 auto sparse_float_shape2 = TensorShape({8, 2}); in TEST_F() local59 sparse_float_shape2, &sparse_float_tensor2)); in TEST_F()
44 auto sparse_float_shape2 = test::AsTensor<int64>({2, 1}); in DecisionTreeTest() local61 {sparse_float_shape1, sparse_float_shape2, multi_sparse_float_shape}, in DecisionTreeTest()