Searched defs:reduced_shape (Results 1 – 4 of 4) sorted by relevance
71 int64 reduced_shape[3] = {1, 1, 1}; in Compute() local
43 TensorShape reduced_shape; member
4134 def reduced_shape(input_shape, axes): function
4871 Shape reduced_shape = ShapeUtil::MakeShapeWithDescendingLayout( in EmitHlo021Tile() local