Home
last modified time | relevance | path

Searched refs:f32_2x2_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_assignment_test.cc42 Shape f32_2x2_ = ShapeUtil::MakeShape(F32, {2, 2}); member in xla::gpu::StreamAssignmentTest
48 /*parameter_number=*/0, f32_2x2_, /*name=*/"x")); in TEST_F()
50 /*parameter_number=*/1, f32_2x2_, /*name=*/"y")); in TEST_F()
52 /*parameter_number=*/2, f32_2x2_, /*name=*/"z")); in TEST_F()
54 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, x, y)); in TEST_F()
56 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, dot1, z)); in TEST_F()
69 /*parameter_number=*/0, f32_2x2_, /*name=*/"x")); in TEST_F()
71 /*parameter_number=*/1, f32_2x2_, /*name=*/"y")); in TEST_F()
73 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, x, y)); in TEST_F()
75 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, y, x)); in TEST_F()
[all …]
Dgpu_hlo_schedule_test.cc39 Shape f32_2x2_ = ShapeUtil::MakeShape(F32, {2, 2}); member in xla::gpu::GpuHloScheduleTest
72 /*parameter_number=*/0, f32_2x2_, /*name=*/"x")); in TEST_F()
74 /*parameter_number=*/1, f32_2x2_, /*name=*/"y")); in TEST_F()
76 /*parameter_number=*/2, f32_2x2_, /*name=*/"z")); in TEST_F()
78 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, x, y)); in TEST_F()
80 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, dot1, z)); in TEST_F()
130 /*parameter_number=*/0, f32_2x2_, /*name=*/"x")); in TEST_F()
132 /*parameter_number=*/1, f32_2x2_, /*name=*/"y")); in TEST_F()
134 /*parameter_number=*/2, f32_2x2_, /*name=*/"z")); in TEST_F()
136 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, x, y)); in TEST_F()
[all …]