Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc152 TF_RETURN_IF_ERROR(constraints->SetOperandLayout(lhs_shape, instr, 0)); in AddBackendConstraintsToDnnConvCustomCall()
153 TF_RETURN_IF_ERROR(constraints->SetOperandLayout(rhs_shape, instr, 1)); in AddBackendConstraintsToDnnConvCustomCall()
168 TF_RETURN_IF_ERROR(constraints->SetOperandLayout(*output_shape, instr, 3)); in AddBackendConstraintsToDnnConvCustomCall()
212 constraints->SetOperandLayout(op0_shape, instruction, 0)); in AddBackendConstraints()
214 constraints->SetOperandLayout(op1_shape, instruction, 1)); in AddBackendConstraints()
227 constraints->SetOperandLayout(shape, instruction, i)); in AddBackendConstraints()
264 constraints->SetOperandLayout(op0_shape, instruction, 0)); in AddBackendConstraints()
266 constraints->SetOperandLayout(op1_shape, instruction, 1)); in AddBackendConstraints()
319 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in PropagateOperandConstraint()
342 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in PropagateBufferConstraint()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc120 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddBackendConstraints()
126 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddBackendConstraints()
145 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddBackendConstraints()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc198 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout() function in xla::LayoutConstraints
259 return SetOperandLayout(shape, instruction, operand_no, mandatory, dfs); in SetArrayOperandLayout()
453 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
473 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
543 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
587 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
636 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
Dlayout_assignment.h170 Status SetOperandLayout(const Shape& shape_with_layout,