HomeSort by relevance Sort by last modified time
    Searched defs:fusion_instruction (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 286 HloInstruction* fusion_instruction = local
290 computation->ReplaceInstruction(add_result, fusion_instruction));
293 fusion_instruction->fused_instructions_computation()->root_instruction();
294 HloInstruction* fused_dot = fusion_instruction->FuseInstruction(dot_result);
308 fusion_instruction->operand(fused_dot->operand(0)->parameter_number());
310 fusion_instruction->operand(fused_dot->operand(1)->parameter_number());
311 result.addend_fusion_param = fusion_instruction->operand(
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_computation.cc 66 HloInstruction* root_instruction, HloInstruction* fusion_instruction)
69 fusion_instruction_(fusion_instruction) {
411 HloInstruction* fusion_instruction) {
432 proto.name(), parameter_count, &instructions, root, fusion_instruction));
437 HloInstruction* fusion_instruction) {
438 CHECK_EQ(HloOpcode::kFusion, fusion_instruction->opcode());
440 TF_CHECK_OK(root->ReplaceAllUsesWith(fusion_instruction));
442 set_root_instruction(fusion_instruction);
447 fusion_instruction->FuseInstruction(instruction);
458 HloInstruction* fusion_instruction = AddInstruction local
    [all...]

Completed in 308 milliseconds