Searched refs:set_element_size_in_bits (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_test.cc | 43 Layout({1, 0}, {Tile({2, 55})}).set_element_size_in_bits(42).ToString(), in TEST_F()
46 Layout({1, 0}, {Tile({-2, 55})}).set_element_size_in_bits(42).ToString(), in TEST_F()
81 EXPECT_EQ(Layout({0, 1, 2}).set_element_size_in_bits(33), in TEST_F()
82 Layout({0, 1, 2}).set_element_size_in_bits(33)); in TEST_F()
83 EXPECT_NE(Layout({0, 1, 2}).set_element_size_in_bits(33), in TEST_F()
84 Layout({0, 1, 2}).set_element_size_in_bits(7)); in TEST_F()
111 expect_unchanged(Layout({0, 1}).set_element_size_in_bits(42)); in TEST_F()
|
| D | layout.cc | 59 layout.set_element_size_in_bits(proto.element_size_in_bits()); in CreateFromProto()
74 proto.set_element_size_in_bits(element_size_in_bits()); in ToProto()
|
| D | layout.h | 192 Layout& set_element_size_in_bits(int64 value) { in set_element_size_in_bits() function
|
| D | layout_util_test.cc | 365 shape.mutable_layout()->set_element_size_in_bits(32); in TEST_F()
371 shape.mutable_layout()->set_element_size_in_bits(32); in TEST_F()
|
| D | layout_util.cc | 74 layout.set_element_size_in_bits(element_size_in_bits); in MakeLayout()
|