Searched refs:body_t (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | while_test.cc | 964 auto body_t = Parameter(&body, 0, tuple_shape, "t"); in XLA_TEST_F() local
965 auto e = GetTupleElement(body_t, 1); in XLA_TEST_F()
1018 auto body_t = Parameter(&body, 0, element_shape, "t"); in XLA_TEST_F() local
1019 auto tuple = Tuple(&body, {body_t, Add(body_t, ConstantR0<float>(&body, 1))}); in XLA_TEST_F()
1056 auto body_t = Parameter(&body, 0, result_shape, "t"); in XLA_TEST_F() local
1058 Tuple(&body, {Add(GetTupleElement(body_t, 0), ConstantR0<int32>(&body, 1)), in XLA_TEST_F()
1059 Add(GetTupleElement(body_t, 1), ConstantR0<int32>(&body, 1))}); in XLA_TEST_F()
|