Searched refs:target_op (Results 1 – 3 of 3) sorted by relevance
363 target_op = None371 target_op = curr_op374 assert target_op380 % target_op.name)
3124 HloInstruction* target_op, in EmitTargetElementLoop() argument3126 return EmitTargetElementLoop(target_op, /*desc=*/"", element_generator); in EmitTargetElementLoop()3130 HloInstruction* target_op, absl::string_view desc, in EmitTargetElementLoop() argument3132 VLOG(2) << "EmitTargetElementLoop: " << target_op->ToString(); in EmitTargetElementLoop()3134 const Shape& target_shape = target_op->shape(); in EmitTargetElementLoop()3135 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(target_op)); in EmitTargetElementLoop()3136 llvm_ir::IrArray target_array = GetIrArrayFor(target_op); in EmitTargetElementLoop()3139 (target_op->opcode() == HloOpcode::kFusion || in EmitTargetElementLoop()3140 target_op->opcode() == HloOpcode::kReduce || in EmitTargetElementLoop()3141 target_op->opcode() == HloOpcode::kReduceWindow)) { in EmitTargetElementLoop()[all …]
312 HloInstruction* target_op,315 HloInstruction* target_op, absl::string_view desc,