Searched refs:SetInstructionLayout (Results 1 – 4 of 4) sorted by relevance
239 constraints->SetInstructionLayout(output_shape, instruction)); in AddBackendConstraints()249 constraints->SetInstructionLayout(output_shape, instruction)); in AddBackendConstraints()301 constraints->SetInstructionLayout(output_shape, instruction)); in AddBackendConstraints()318 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in PropagateOperandConstraint()
297 Status LayoutConstraints::SetInstructionLayout( in SetInstructionLayout() function in xla::LayoutConstraints465 constraints->SetInstructionLayout(instruction->shape(), instruction)); in AddMandatoryConstraints()478 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddMandatoryConstraints()485 constraints->SetInstructionLayout(custom_call->shape(), custom_call)); in AddMandatoryConstraints()507 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddMandatoryConstraints()526 constraints->SetInstructionLayout(instruction->shape(), instruction)); in AddMandatoryConstraints()542 constraints->SetInstructionLayout(new_buffer_shape, instruction)); in AddMandatoryConstraints()556 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddMandatoryConstraints()605 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddMandatoryConstraints()648 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddMandatoryConstraints()
187 Status SetInstructionLayout(const Shape& shape_with_layout,
117 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddBackendConstraints()