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.cc454 const Shape& reference_shape = operands[0]->shape(); in HandleConcatenate() local
455 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()