Searched refs:reference_shape (Results 1 – 1 of 1) sorted by relevance
2236 const Shape& reference_shape = operands[0]->shape(); in HandleConcatenate() local2237 CHECK(!ShapeUtil::IsTuple(reference_shape)); in HandleConcatenate()2238 const int64 rank = ShapeUtil::Rank(reference_shape); in HandleConcatenate()2243 DimensionVector concat_dimensions(reference_shape.dimensions().begin(), in HandleConcatenate()2244 reference_shape.dimensions().end()); in HandleConcatenate()2256 reference_shape.element_type(), concat_dimensions); in HandleConcatenate()