Searched defs:r2f32 (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | algebraic_simplifier_test.cc | 722 Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST_F() local 758 Shape r2f32 = ShapeUtil::MakeShape(F32, {32, 1}); in TEST_F() local 805 Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST_F() local 1053 Shape r2f32 = ShapeUtil::MakeShape(F32, {42, 123}); in TEST_F() local 1286 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 1307 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 1331 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 1355 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 2752 Shape r2f32 = ShapeUtil::MakeShape(F32, {100, 99}); in TEST_F() local 5084 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local [all …]
|
H A D | hlo_cost_analysis_test.cc | 694 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 761 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
|
H A D | hlo_instruction_test.cc | 1181 const Shape r2f32 = ShapeUtil::MakeShapeWithLayout(F32, {10, 10}, {1, 0}); in TEST_F() local 1198 const Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 5}); in TEST_F() local
|
H A D | heap_simulator_test.cc | 133 const Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
|