Searched refs:release_tensor (Results 1 – 7 of 7) sorted by relevance
189 PROTOBUF_NODISCARD ::tensorflow::TensorProto* release_tensor(); in Swap()295 inline ::tensorflow::TensorProto* NamedTensorProto::release_tensor() { in release_tensor() function
312 PROTOBUF_NODISCARD ::tensorflow::TensorProto* release_tensor(); in Swap()4019 PROTOBUF_NODISCARD ::tensorflow::TensorProto* release_tensor(); in Swap()4487 inline ::tensorflow::TensorProto* Operation_Input::release_tensor() { in release_tensor() function6920 inline ::tensorflow::TensorProto* SendPackedHandleOp_LocalTensorHandle::release_tensor() { in release_tensor() function
3311 PROTOBUF_NODISCARD ::tensorflow::TensorProto* release_tensor(); in Swap()8560 inline ::tensorflow::TensorProto* RecvTensorResponse::release_tensor() { in release_tensor() function
605 Tensor release_tensor(DT_INT64, TensorShape({1})); in TEST() local606 release_tensor.flat<int64_t>()(0) = allocation_handle; in TEST()608 auto release = ops::XRTReleaseAllocationHandle(root, release_tensor); in TEST()637 Tensor release_tensor(DT_INT64, TensorShape({2})); in TEST() local638 release_tensor.flat<int64_t>()(0) = allocation_handle1; in TEST()639 release_tensor.flat<int64_t>()(1) = allocation_handle2; in TEST()641 auto release = ops::XRTReleaseAllocationHandle(root, release_tensor); in TEST()686 Tensor release_tensor(DT_INT64, TensorShape({2})); in TEST() local687 release_tensor.flat<int64_t>()(0) = compilation_handle1; in TEST()688 release_tensor.flat<int64_t>()(1) = compilation_handle2; in TEST()[all …]
360 PROTOBUF_NODISCARD ::tensorflow::TensorDescription* release_tensor(); in Swap()681 PROTOBUF_NODISCARD ::tensorflow::TensorDescription* release_tensor(); in Swap()1283 inline ::tensorflow::TensorDescription* MemoryLogTensorAllocation::release_tensor() { in release_tensor() function1521 inline ::tensorflow::TensorDescription* MemoryLogTensorOutput::release_tensor() { in release_tensor() function
606 PROTOBUF_NODISCARD ::tensorflow::TensorProto* release_tensor(); in Swap()1595 inline ::tensorflow::TensorProto* AttrValue::release_tensor() { in release_tensor() function
1443 PROTOBUF_NODISCARD ::tensorflow::TensorProto* release_tensor(); in Swap()2976 inline ::tensorflow::TensorProto* Summary_Value::release_tensor() { in release_tensor() function