Searched refs:SetInstructionLayout (Results 1 – 4 of 4) sorted by relevance
216 constraints->SetInstructionLayout(output_shape, instruction)); in AddBackendConstraints()268 constraints->SetInstructionLayout(output_shape, instruction)); in AddBackendConstraints()285 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in PropagateOperandConstraint()
287 Status LayoutConstraints::SetInstructionLayout( in SetInstructionLayout() function in xla::LayoutConstraints449 constraints->SetInstructionLayout(instruction->shape(), instruction)); in AddMandatoryConstraints()463 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddMandatoryConstraints()471 constraints->SetInstructionLayout(custom_call->shape(), custom_call)); in AddMandatoryConstraints()493 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddMandatoryConstraints()524 constraints->SetInstructionLayout(new_buffer_shape, instruction)); in AddMandatoryConstraints()538 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddMandatoryConstraints()585 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddMandatoryConstraints()633 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddMandatoryConstraints()
185 Status SetInstructionLayout(const Shape& shape_with_layout,
117 TF_RETURN_IF_ERROR(constraints->SetInstructionLayout( in AddBackendConstraints()