Home
last modified time | relevance | path

Searched refs:shape_in_default_mem (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_test.cc3657 Shape shape_in_default_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P() local
3671 EXPECT_THAT(tanh0, op::ShapeWithLayout(shape_in_default_mem)); in TEST_P()
3672 EXPECT_THAT(tanh1, op::ShapeWithLayout(shape_in_default_mem)); in TEST_P()
3673 EXPECT_THAT(tanh2, op::ShapeWithLayout(shape_in_default_mem)); in TEST_P()
3674 EXPECT_THAT(tanh3, op::ShapeWithLayout(shape_in_default_mem)); in TEST_P()
3675 EXPECT_THAT(tanh4, op::ShapeWithLayout(shape_in_default_mem)); in TEST_P()
3765 Shape shape_in_default_mem = ShapeUtil::MakeShapeWithLayout( in TEST_P() local