Searched defs:int_shape (Results 1 – 3 of 3) sorted by relevance
32 Shape int_shape = ShapeUtil::MakeShape(xla::S32, {}); in XLA_TEST_F() local
158 def _get_shape_tuple(self, init_tuple, tensor, start_idx, int_shape=None): argument
943 def int_shape(x): function