Searched defs:truth_shape (Results 1 – 3 of 3) sorted by relevance
42 const Tensor& truth_shape) { in ValidateShapes()109 const Tensor* truth_shape; in Compute() local
103 truth_shape, argument
1230 Tensor truth_shape = test::AsTensor<int64>({20, 3, 40, 5}); in TEST() local