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.cc216 const OperandLayoutConstraint* curr_shape_layout = in SetOperandLayout() local
218 if (curr_shape_layout != nullptr) { in SetOperandLayout()
219 if (curr_shape_layout->shape_layout().MatchesLayoutInShape( in SetOperandLayout()
224 if (curr_shape_layout->mandatory()) { in SetOperandLayout()
229 curr_shape_layout->shape_layout().ToString(), in SetOperandLayout()
277 const ShapeLayout* curr_shape_layout = ResultLayout(); in SetResultLayout() local
278 if (curr_shape_layout != nullptr) { in SetResultLayout()
279 if (!curr_shape_layout->MatchesLayoutInShape( in SetResultLayout()
284 computation_->name(), curr_shape_layout->ToString(), in SetResultLayout()