Searched defs:reduced_shape (Results 1 – 3 of 3) sorted by relevance
73 int64 reduced_shape[3] = {1, 1, 1}; in Compute() local
43 TensorShape reduced_shape; member
3174 def reduced_shape(input_shape, axes): function