Searched refs:fused_param (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dliveness_util.cc46 [=](HloInstruction* fused_param) {
47 return user->operand(fused_param->parameter_number()) == operand;
150 [&](HloInstruction* fused_param) {
151 return fusion->operand(fused_param->parameter_number()) == operand;
156 auto* fused_param = *fused_param_it; local
159 fused_param, operand_index, points_to_analysis);
H A Dhlo_verifier.cc694 for (auto fused_param : fused_parameters) {
695 int64 param_no = fused_param->parameter_number();
713 if (!ShapeUtil::Compatible(fused_param->shape(),
H A Dhlo_instruction.cc1000 HloInstruction* fused_param = nullptr; local
1003 fused_param = fused_parameters[i];
1008 if (fused_param == nullptr) {
1012 fused_param = AddFusionOperand(operand);
1014 TF_CHECK_OK(clone->ReplaceOperandWith(operand_num, fused_param));

Completed in 179 milliseconds