Searched refs:reference_shape (Results 1 – 1 of 1) sorted by relevance
454 const Shape& reference_shape = operands[0]->shape(); in HandleConcatenate() local455 CHECK(reference_shape.IsArray()); in HandleConcatenate()456 const int64 rank = reference_shape.rank(); in HandleConcatenate()461 DimensionVector concat_dimensions(reference_shape.dimensions().begin(), in HandleConcatenate()462 reference_shape.dimensions().end()); in HandleConcatenate()474 reference_shape.element_type(), concat_dimensions); in HandleConcatenate()