Searched refs:allocation_handle (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 331 int64 allocation_handle = handle_tensor.scalar<int64>()(); in Compute() local
347 ctx, XRTTupleAllocation::Lookup(rm, allocation_handle, &allocation)); in Compute()
351 VLOG(2) << "Releasing handle " << allocation_handle; in Compute()
353 rm, allocation_handle)); in Compute()
471 int64 allocation_handle = handle_tensor.scalar<int64>()(); in Compute() local
478 ctx, XRTTupleAllocation::Lookup(rm, allocation_handle, &allocation)); in Compute()
482 VLOG(2) << "Releasing handle " << allocation_handle; in Compute()
484 rm, allocation_handle)); in Compute()
527 int64 allocation_handle = handle_tensor.scalar<int64>()(); in Compute() local
534 ctx, XRTTupleAllocation::Lookup(rm, allocation_handle, &allocation)); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xrt.cc | 53 auto allocation_handle = in ~XrtAllocation() local
56 tensorflow::ops::XRTReleaseAllocationHandle(root, allocation_handle); in ~XrtAllocation()
64 inputs.insert({allocation_handle, handle()}); in ~XrtAllocation()
101 auto allocation_handle = in ToLiteral() local
103 auto read_literal = tensorflow::ops::XRTReadLiteral(root, allocation_handle); in ToLiteral()
108 inputs.insert({allocation_handle, handle()}); in ToLiteral()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 377 int64 allocation_handle = outputs[1].scalar<int64>()(); in TEST() local
388 ops::XRTWriteLiteral(root, Input(allocation_handle), new_value); in TEST()
392 EXPECT_EQ(allocation_handle, outputs[0].scalar<int64>()()); in TEST()
395 auto read_after_write = ops::XRTReadLiteral(root, Input(allocation_handle)); in TEST()
404 release_tensor.flat<int64>()(0) = allocation_handle; in TEST()
519 int64 allocation_handle = outputs[0].scalar<int64>()(); in TEST() local
528 auto read_after_clear = ops::XRTReadLiteral(root, Input(allocation_handle)); in TEST()
|