Searched refs:reference_shape (Results 1 – 1 of 1) sorted by relevance
491 const Shape& reference_shape = operands[0]->shape(); in HandleConcatenate() local492 CHECK(reference_shape.IsArray()); in HandleConcatenate()493 const int64 rank = reference_shape.rank(); in HandleConcatenate()498 DimensionVector concat_dimensions(reference_shape.dimensions().begin(), in HandleConcatenate()499 reference_shape.dimensions().end()); in HandleConcatenate()511 reference_shape.element_type(), concat_dimensions); in HandleConcatenate()