Searched refs:SetOperandLayout (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_layout_assignment.cc | 198 TF_RETURN_IF_ERROR(SetOperandLayout(lhs_shape, instr, 0)); in AddBackendConstraintsToDnnConvCustomCall()
199 TF_RETURN_IF_ERROR(SetOperandLayout(rhs_shape, instr, 1)); in AddBackendConstraintsToDnnConvCustomCall()
213 TF_RETURN_IF_ERROR(SetOperandLayout(*output_shape, instr, 3)); in AddBackendConstraintsToDnnConvCustomCall()
325 SetOperandLayout(shape, instruction, /*operand_no=*/0)); in AddBackendConstraints()
333 TF_RETURN_IF_ERROR(SetOperandLayout(op0_shape, instruction, 0)); in AddBackendConstraints()
344 TF_RETURN_IF_ERROR(SetOperandLayout(shape, instruction, i)); in AddBackendConstraints()
370 TF_RETURN_IF_ERROR(SetOperandLayout(op0_shape, instruction, 0)); in AddBackendConstraints()
371 TF_RETURN_IF_ERROR(SetOperandLayout(op1_shape, instruction, 1)); in AddBackendConstraints()
411 return SetOperandLayout(shape, instruction, operand); in SetDotOperandLayout()
416 return SetOperandLayout(shape, instruction, operand); in SetDotOperandLayout()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_layout_assignment.cc | 121 TF_RETURN_IF_ERROR(SetOperandLayout( in AddBackendConstraints()
128 SetOperandLayout(ColMajorShape(op->shape()), instruction, *op_idx)); in AddBackendConstraints()
147 SetOperandLayout(operand_shape, instruction, operand_no)); in AddBackendConstraints()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_assignment.h | 374 Status SetOperandLayout(const Shape& shape_with_layout,
377 return SetOperandLayout(shape_with_layout, instruction, operand_no,
380 Status SetOperandLayout(const Shape& shape_with_layout,
|
| H A D | layout_assignment.cc | 235 Status LayoutAssignment::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout() function in xla::LayoutAssignment
338 return SetOperandLayout(shape, instruction, operand_no, mandatory, dfs, in SetArrayOperandLayout()
564 TF_RETURN_IF_ERROR(constraints->SetOperandLayout( in PropagateParameterLayoutToUsers()
599 TF_RETURN_IF_ERROR(SetOperandLayout(instruction->outfeed_shape(), in AddMandatoryConstraints()
633 TF_RETURN_IF_ERROR(SetOperandLayout( in AddMandatoryConstraints()
676 TF_RETURN_IF_ERROR(SetOperandLayout( in AddMandatoryConstraints()
734 SetOperandLayout(body_layout.result_shape(), instruction, 0)); in AddMandatoryConstraints()
777 TF_RETURN_IF_ERROR(SetOperandLayout( in AddMandatoryConstraints()
783 SetOperandLayout(best_branch_computation_layout.parameter_shape(0), in AddMandatoryConstraints()
|