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

  /external/tensorflow/tensorflow/compiler/xla/service/
instruction_fusion_test.cc 117 auto small_shape = ShapeUtil::MakeShape(F32, {16}); local
120 HloInstruction::CreateParameter(0, small_shape, "0"));
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_instruction_fusion_test.cc 569 Shape small_shape = ShapeUtil::MakeShape(F32, {1, 4}); local
576 /*parameter_number=*/0, small_shape, "param"));
578 HloInstruction::CreateUnary(small_shape, HloOpcode::kExp, small_param));
594 Shape small_shape = ShapeUtil::MakeShape(F32, {1, 4}); local
601 /*parameter_number=*/0, small_shape, "param"));
606 HloInstruction::CreateUnary(small_shape, HloOpcode::kExp, small_param));

Completed in 1802 milliseconds