Searched refs:MatchesLayoutInShape (Results 1 – 4 of 4) sorted by relevance
48 bool MatchesLayoutInShape(const Shape& shape) const;
49 bool ShapeLayout::MatchesLayoutInShape(const Shape& shape) const { in MatchesLayoutInShape() function in xla::ShapeLayout
209 if (curr_shape_layout->shape_layout().MatchesLayoutInShape( in SetOperandLayout()269 if (!curr_shape_layout->MatchesLayoutInShape(shape_with_layout)) { in SetResultLayout()662 TF_RET_CHECK(computation_layout.parameter_layout(i).MatchesLayoutInShape( in CheckCallLayout()666 computation_layout.result_layout().MatchesLayoutInShape(call->shape())); in CheckCallLayout()696 condition_computation_layout.parameter_layout(0).MatchesLayoutInShape( in CheckWhileLayout()698 TF_RET_CHECK(body_computation_layout.parameter_layout(0).MatchesLayoutInShape( in CheckWhileLayout()701 body_computation_layout.result_layout().MatchesLayoutInShape(init_shape)); in CheckWhileLayout()715 branch_computation_layouts[j].result_layout().MatchesLayoutInShape( in CheckConditionalLayout()718 branch_computation_layouts[j].result_layout().MatchesLayoutInShape( in CheckConditionalLayout()721 branch_computation_layouts[j].parameter_layout(0).MatchesLayoutInShape( in CheckConditionalLayout()[all …]
67 if (!computation_layout.parameter_layout(i).MatchesLayoutInShape( in ValidateExecutionOptions()