algebraic_simplifier_test.cc | 358 Shape r1f32 = ShapeUtil::MakeShape(F32, {7}); local 361 HloInstruction::CreateParameter(0, r1f32, "param0")); 363 HloInstruction::CreateParameter(1, r1f32, "param1")); 367 HloInstruction::CreateBinary(r1f32, HloOpcode::kPower, param1, param2)); 369 HloInstruction::CreateBinary(r1f32, HloOpcode::kDivide, param0, power)); 391 Shape r1f32 = ShapeUtil::MakeShape(F32, {3}); local 394 HloInstruction::CreateParameter(0, r1f32, "param0")); 398 builder.AddInstruction(HloInstruction::CreateBinary(r1f32, HloOpcode::kDivide, 414 Shape r1f32 = ShapeUtil::MakeShape(F32, {7}); local 417 HloInstruction::CreateParameter(0, r1f32, "param0")) 785 Shape r1f32 = ShapeUtil::MakeShape(F32, {42}); local 1059 Shape r1f32 = ShapeUtil::MakeShape(F32, {100}); local 1080 Shape r1f32 = ShapeUtil::MakeShape(F32, {kParamLength}); local 1112 Shape r1f32 = ShapeUtil::MakeShape(F32, {kParamLength}); local 1139 Shape r1f32 = ShapeUtil::MakeShape(F32, {100}); local 2062 Shape r1f32 = ShapeUtil::MakeShape(F32, {100}); local 2332 Shape r1f32 = ShapeUtil::MakeShape(F32, {1}); local [all...] |