Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization_test.cc96 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
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.cc433 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
/external/ComputeLibrary/tests/validation/CL/
DScale.cpp210 …_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()
/external/ComputeLibrary/tests/validation/NEON/
DScale.cpp331 …(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()
/external/ComputeLibrary/tests/validation/dynamic_fusion/gpu/cl/
DResize.cpp242 …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()