Home
last modified time | relevance | path

Searched refs:reference_shape (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc491 const Shape& reference_shape = operands[0]->shape(); in HandleConcatenate() local
492 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()