Searched refs:infeed_dequeue_tuple (Results 1 – 3 of 3) sorted by relevance
221 def infeed_dequeue_tuple(dtypes, shapes, name=None): function243 return gen_tpu_ops.infeed_dequeue_tuple(dtypes, shapes, name=name)
506 dequeue_op = tpu_ops.infeed_dequeue_tuple(509 dequeue_op = tpu_ops.infeed_dequeue_tuple(812 values = tpu_ops.infeed_dequeue_tuple(
1075 // CHECK-LABEL: func @infeed_dequeue_tuple1076 func @infeed_dequeue_tuple() -> (tensor<3xi32>, tensor<4xf32>) {