Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc206 const OperandLayoutConstraint* curr_shape_layout = in SetOperandLayout() local
208 if (curr_shape_layout != nullptr) { in SetOperandLayout()
209 if (curr_shape_layout->shape_layout().MatchesLayoutInShape( in SetOperandLayout()
214 if (curr_shape_layout->mandatory()) { in SetOperandLayout()
219 curr_shape_layout->shape_layout().ToString(), in SetOperandLayout()
267 const ShapeLayout* curr_shape_layout = ResultLayout(); in SetResultLayout() local
268 if (curr_shape_layout != nullptr) { in SetResultLayout()
269 if (!curr_shape_layout->MatchesLayoutInShape(shape_with_layout)) { in SetResultLayout()
273 computation_->name(), curr_shape_layout->ToString(), in SetResultLayout()