Searched refs:release_tensor (Results 1 – 1 of 1) sorted by relevance
403 Tensor release_tensor(DT_INT64, TensorShape({1})); in TEST() local404 release_tensor.flat<int64>()(0) = allocation_handle; in TEST()406 auto release = ops::XRTReleaseAllocationHandle(root, release_tensor); in TEST()436 Tensor release_tensor(DT_INT64, TensorShape({2})); in TEST() local437 release_tensor.flat<int64>()(0) = allocation_handle1; in TEST()438 release_tensor.flat<int64>()(1) = allocation_handle2; in TEST()440 auto release = ops::XRTReleaseAllocationHandle(root, release_tensor); in TEST()493 Tensor release_tensor(DT_INT64, TensorShape({2})); in TEST() local494 release_tensor.flat<int64>()(0) = compilation_handle1; in TEST()495 release_tensor.flat<int64>()(1) = compilation_handle2; in TEST()[all …]