Searched refs:init_tuple (Results 1 – 4 of 4) 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/compiler/xla/service/ |
| D | while_util.cc | 256 TF_ASSIGN_OR_RETURN(HloInstruction * init_tuple, in MakeCountedLoop()
262 module->AddEmbeddedComputation(std::move(body)), init_tuple)); in MakeCountedLoop()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | wrappers.py | 140 def _get_shape_tuple(self, init_tuple, tensor, start_idx, int_shape=None): argument
165 return init_tuple + tuple(int_shape)
171 return init_tuple + tuple(int_shape)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 660 auto init_tuple = in CreateWhile32() local
662 auto tuple_type = init_tuple.getType(); in CreateWhile32()
665 auto while_op = builder->create<mhlo::WhileOp>(loc, init_tuple); in CreateWhile32()
|