Searched defs:init_tuple (Results 1 – 2 of 2) sorted by relevance
39 XlaOp init_tuple = Tuple(&b, {num_iter, some_buffer}); in XLA_TEST_F() local
158 def _get_shape_tuple(self, init_tuple, tensor, start_idx, int_shape=None): argument