Home
last modified time | relevance | path

Searched refs:allocation_handle (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h379 int64 allocation_handle = handle_tensor.scalar<int64>()(); in Compute() local
395 OP_REQUIRES_OK(ctx, memory_manager->Lookup(allocation_handle, &allocation)); in Compute()
398 VLOG(2) << "Releasing handle " << allocation_handle; in Compute()
399 OP_REQUIRES_OK(ctx, memory_manager->Release(allocation_handle)); in Compute()
501 int64 allocation_handle = handle_tensor.scalar<int64>()(); in Compute() local
508 OP_REQUIRES_OK(ctx, memory_manager->Lookup(allocation_handle, &allocation)); in Compute()
511 VLOG(2) << "Releasing handle " << allocation_handle; in Compute()
512 OP_REQUIRES_OK(ctx, memory_manager->Release(allocation_handle)); in Compute()
554 int64 allocation_handle = handle_tensor.scalar<int64>()(); in Compute() local
561 OP_REQUIRES_OK(ctx, memory_manager->Lookup(allocation_handle, &allocation)); in Compute()
[all …]
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc582 int64 allocation_handle = outputs[1].scalar<int64>()(); in TEST() local
592 ops::XRTWriteLiteral(root, Input(allocation_handle), new_value); in TEST()
596 EXPECT_EQ(allocation_handle, outputs[0].scalar<int64>()()); in TEST()
598 auto read_after_write = ops::XRTReadLiteral(root, Input(allocation_handle)); in TEST()
607 release_tensor.flat<int64>()(0) = allocation_handle; in TEST()
711 int64 allocation_handle = outputs[0].scalar<int64>()(); in TEST() local
719 auto read_after_clear = ops::XRTReadLiteral(root, Input(allocation_handle)); in TEST()