Searched refs:release_tensor (Results 1 – 1 of 1) sorted by relevance
606 Tensor release_tensor(DT_INT64, TensorShape({1})); in TEST() local607 release_tensor.flat<int64>()(0) = allocation_handle; in TEST()609 auto release = ops::XRTReleaseAllocationHandle(root, release_tensor); in TEST()638 Tensor release_tensor(DT_INT64, TensorShape({2})); in TEST() local639 release_tensor.flat<int64>()(0) = allocation_handle1; in TEST()640 release_tensor.flat<int64>()(1) = allocation_handle2; in TEST()642 auto release = ops::XRTReleaseAllocationHandle(root, release_tensor); in TEST()687 Tensor release_tensor(DT_INT64, TensorShape({2})); in TEST() local688 release_tensor.flat<int64>()(0) = compilation_handle1; in TEST()689 release_tensor.flat<int64>()(1) = compilation_handle2; in TEST()[all …]