Home
last modified time | relevance | path

Searched defs:f32_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization_test.cc94 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
122 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
152 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
249 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
277 Shape f32_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_F() local
307 Shape f32_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_F() local
Dbfloat16_conversion_folding_test.cc85 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
119 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
154 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
189 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
229 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
Dbfloat16_propagation_test.cc901 Shape f32_shape = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local