Searched refs:while_output (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5119 SmallVector<Value, 2> while_output; in matchAndRewrite() local
5121 &while_output, &rewriter); in matchAndRewrite()
5122 Value swaped_indices = while_output[1]; in matchAndRewrite()
5968 SmallVector<Value, 4> while_output; in QRBlock() local
5970 &while_output, rewriter); in QRBlock()
5971 *r = while_output[0]; in QRBlock()
5972 *vs = while_output[1]; in QRBlock()
5973 *taus = while_output[2]; in QRBlock()
6073 SmallVector<Value, 4> while_output; in ComputeWYRepresentation() local
6074 CreateWhile32(loc, n - 1, body_fn, {w, vs, taus}, &while_output, rewriter); in ComputeWYRepresentation()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 4672 while_output = self._buildWhileWithShapeInvariants(
4674 self.assertIsNone(while_output.shape.rank)
4680 while_output = self._buildWhileWithShapeInvariants(
4682 self.assertAllEqual(while_output.shape.as_list(), [None])
|