Searched refs:init_tuple (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | outfeed_in_nested_computation_test.cc | 39 XlaOp init_tuple = Tuple(&b, {num_iter, some_buffer}); in XLA_TEST_F() local
69 XlaOp result_tuple = While(loop_cond, loop_body, init_tuple); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | wrappers.py | 158 def _get_shape_tuple(self, init_tuple, tensor, start_idx, int_shape=None): argument
183 return init_tuple + tuple(int_shape)
189 return init_tuple + tuple(int_shape)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_util.cc | 241 TF_ASSIGN_OR_RETURN(HloInstruction * init_tuple, in MakeCountedLoop()
247 module->AddEmbeddedComputation(std::move(body)), init_tuple)); in MakeCountedLoop()
|