Searched refs:StoreComputationSnapshot (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 354 void StoreComputationSnapshot(const xla::XlaComputation& computation, in StoreComputationSnapshot() function
656 StoreComputationSnapshot(AddAndScale(), c1.mutable_hlo_snapshot()); in TEST()
668 StoreComputationSnapshot(AddAndTuple(), c2.mutable_hlo_snapshot()); in TEST()
893 StoreComputationSnapshot(fn(), c.mutable_hlo_snapshot()); in TEST()
963 StoreComputationSnapshot(fn(), c.mutable_hlo_snapshot()); in TEST()
1085 StoreComputationSnapshot(AddAndScale(), c.mutable_hlo_snapshot()); in TEST()
1142 StoreComputationSnapshot(ReturnDynamicR1(), c.mutable_hlo_snapshot()); in TEST()
1199 StoreComputationSnapshot(ReturnDynamicR2(), c.mutable_hlo_snapshot()); in TEST()
1254 StoreComputationSnapshot(ReturnDynamicR1Tuple(), c.mutable_hlo_snapshot()); in TEST()
1317 StoreComputationSnapshot(AcceptDynamicR1Tuple(), c.mutable_hlo_snapshot()); in TEST()
[all …]
|