| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | bfloat16_normalization_test.cc | 96 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
124 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
251 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
281 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
310 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
338 Shape f32_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_F() local
367 Shape f32_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_F() local
|
| H A D | bfloat16_conversion_folding_test.cc | 85 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
|
| H A D | bfloat16_propagation_test.cc | 433 Shape f32_shape = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local
888 Shape f32_shape = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/CL/ |
| H A D | Scale.cpp | 210 …_CASE(Run, CLScaleFixture<float>, framework::DatasetMode::ALL, ASSEMBLE_DATASET(f32_shape, ScaleSa… in FIXTURE_DATA_TEST_CASE()
219 …aleMixedDataLayoutFixture<float>, framework::DatasetMode::ALL, ASSEMBLE_DATASET(f32_shape, ScaleSa… in FIXTURE_DATA_TEST_CASE()
228 …gnCorners, CLScaleFixture<float>, framework::DatasetMode::ALL, ASSEMBLE_DATASET(f32_shape, ScaleAl… in FIXTURE_DATA_TEST_CASE()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/ |
| H A D | Scale.cpp | 331 …(RunSmall, NEScaleFixture<float>, framework::DatasetMode::ALL, ASSEMBLE_DATASET(f32_shape, ScaleSa… in FIXTURE_DATA_TEST_CASE()
340 …edDataLayoutFixture<float>, framework::DatasetMode::PRECOMMIT, ASSEMBLE_DATASET(f32_shape, ScaleSa… in FIXTURE_DATA_TEST_CASE()
349 …gnCorners, NEScaleFixture<float>, framework::DatasetMode::ALL, ASSEMBLE_DATASET(f32_shape, ScaleAl… in FIXTURE_DATA_TEST_CASE()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/dynamic_fusion/gpu/cl/ |
| H A D | Resize.cpp | 242 …usionResizeFixture<float>, framework::DatasetMode::ALL, ASSEMBLE_DATASET_DYNAMIC_FUSION(f32_shape,… in FIXTURE_DATA_TEST_CASE()
252 …usionResizeFixture<float>, framework::DatasetMode::ALL, ASSEMBLE_DATASET_DYNAMIC_FUSION(f32_shape,… in FIXTURE_DATA_TEST_CASE()
|