Searched defs:reduced_shape (Results 1 – 5 of 5) sorted by relevance
229 auto reduced_shape = ReducedShapeHelper(scope, input_shape, reduced_axes); in EinsumGradReducedHelper() local250 auto reduced_shape = Concat( in EinsumGradReducedHelper() local
71 int64_t reduced_shape[3] = {1, 1, 1}; in Compute() local
45 TensorShape reduced_shape; member
4500 def reduced_shape(input_shape, axes): function
4239 Shape reduced_shape = ShapeUtil::MakeShapeWithDescendingLayout( in EmitHlo021Tile() local