Searched refs:StoreComputationSnapshot (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 218 void StoreComputationSnapshot(const xla::XlaComputation& computation, in StoreComputationSnapshot() function
456 StoreComputationSnapshot(AddAndScale(), c1.mutable_hlo_snapshot()); in TEST()
468 StoreComputationSnapshot(AddAndTuple(), c2.mutable_hlo_snapshot()); in TEST()
704 StoreComputationSnapshot(AddAndScale(), c.mutable_hlo_snapshot()); in TEST()
757 StoreComputationSnapshot(AddAndScale(), c.mutable_hlo_snapshot()); in TEST()
823 StoreComputationSnapshot(xla_computation, c.mutable_hlo_snapshot()); in TEST()
878 StoreComputationSnapshot(Dot(), c.mutable_hlo_snapshot()); in TEST()
923 StoreComputationSnapshot(OnePlusTwo(), c.mutable_hlo_snapshot()); in TEST()
963 StoreComputationSnapshot(AddAndTuple(), c.mutable_hlo_snapshot()); in TEST()
1014 StoreComputationSnapshot(AddAndSubTuple(), c.mutable_hlo_snapshot()); in TEST()
[all …]
|