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.cc174 TF_RETURN_IF_ERROR(constraints->SetOperandLayout(lhs_shape, instr, 0)); in AddBackendConstraintsToDnnConvCustomCall()
175 TF_RETURN_IF_ERROR(constraints->SetOperandLayout(rhs_shape, instr, 1)); in AddBackendConstraintsToDnnConvCustomCall()
190 TF_RETURN_IF_ERROR(constraints->SetOperandLayout(*output_shape, instr, 3)); in AddBackendConstraintsToDnnConvCustomCall()
235 constraints->SetOperandLayout(op0_shape, instruction, 0)); in AddBackendConstraints()
237 constraints->SetOperandLayout(op1_shape, instruction, 1)); in AddBackendConstraints()
247 constraints->SetOperandLayout(op0_shape, instruction, 0)); in AddBackendConstraints()
260 constraints->SetOperandLayout(shape, instruction, i)); in AddBackendConstraints()
297 constraints->SetOperandLayout(op0_shape, instruction, 0)); in AddBackendConstraints()
299 constraints->SetOperandLayout(op1_shape, instruction, 1)); in AddBackendConstraints()
352 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in PropagateOperandConstraint()
[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.cc208 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout() function in xla::LayoutConstraints
269 return SetOperandLayout(shape, instruction, operand_no, mandatory, dfs); in SetArrayOperandLayout()
469 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
487 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
561 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
603 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
639 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
644 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in AddMandatoryConstraints()
Dlayout_assignment.h172 Status SetOperandLayout(const Shape& shape_with_layout,