Searched defs:range_shape (Results 1 – 2 of 2) sorted by relevance
143 const TensorShape range_shape = in TEST_P() local194 const TensorShape range_shape = in TEST_P() local248 const TensorShape range_shape = in TEST_P() local331 const TensorShape range_shape = in TEST_P() local386 const TensorShape range_shape = in TEST_P() local
113 auto range_shape = (axis_ == -1) ? TensorShape({}) : TensorShape({depth}); in Compute() local299 auto range_shape = (axis_ == -1) ? TensorShape({}) : TensorShape({depth}); in Compute() local