Searched refs:set_element_size_in_bits (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_test.cc | 41 Layout({1, 0}, {Tile({2, 55})}).set_element_size_in_bits(42).ToString(), in TEST_F()
48 Layout({1, 0}, {Tile({-2, 55})}).set_element_size_in_bits(42).ToString(), in TEST_F()
78 EXPECT_EQ(Layout({0, 1, 2}).set_element_size_in_bits(33), in TEST_F()
79 Layout({0, 1, 2}).set_element_size_in_bits(33)); in TEST_F()
80 EXPECT_NE(Layout({0, 1, 2}).set_element_size_in_bits(33), in TEST_F()
81 Layout({0, 1, 2}).set_element_size_in_bits(7)); in TEST_F()
107 expect_unchanged(Layout({0, 1}).set_element_size_in_bits(42)); in TEST_F()
|
| D | layout.cc | 58 layout.set_element_size_in_bits(proto.element_size_in_bits()); in CreateFromProto()
73 proto.set_element_size_in_bits(element_size_in_bits()); in ToProto()
|
| D | layout.h | 207 Layout& set_element_size_in_bits(int64 value) { in set_element_size_in_bits() function
|
| D | layout_util_test.cc | 317 shape.mutable_layout()->set_element_size_in_bits(32); in TEST_F()
323 shape.mutable_layout()->set_element_size_in_bits(32); in TEST_F()
|
| D | layout_util.cc | 76 layout.set_element_size_in_bits(element_size_in_bits); in MakeLayout()
|
| D | shape_util.cc | 339 new_shape.mutable_layout()->set_element_size_in_bits( in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 285 new_shape.mutable_layout()->set_element_size_in_bits(0); in MakeFakeLiteralInternal()
387 new_shape.mutable_layout()->set_element_size_in_bits(0); in MakeFakeLiteralInternalWithBounds()
|