Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization_test.cc95 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
123 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
153 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
250 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
278 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {1024}); in TEST_F() local
308 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {1024}); in TEST_F() local
343 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {4, 4}); in TEST_F() local
Dbfloat16_conversion_folding_test.cc86 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
120 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
155 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
190 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
230 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
Dbfloat16_propagation_test.cc902 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {4, 4}); in TEST_F() local