Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_dyn_shape_test.cc28 xla::Shape dyn_input_shape = xla::ShapeUtil::MakeShape(xla::F32, {2, 4}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_dyn_shape_test.cc31 xla::Shape dyn_input_shape = xla::ShapeUtil::MakeShape(xla::F32, {2, 4}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc1309 xla::Shape dyn_input_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in TEST() local
1363 xla::Shape dyn_input_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in TEST() local