Home
last modified time | relevance | path

Searched refs:TFE_DeleteTensorHandle (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc67 TFE_DeleteTensorHandle(m); in BM_InitOp()
108 TFE_DeleteTensorHandle(m); in BM_Execute()
147 TFE_DeleteTensorHandle(m); in BM_Execute_Identity()
196 TFE_DeleteTensorHandle(h); in TEST()
239 TFE_DeleteTensorHandle(hdevice); in TensorHandleCopyBetweenDevices()
247 TFE_DeleteTensorHandle(hdevice2); in TensorHandleCopyBetweenDevices()
251 TFE_DeleteTensorHandle(hcopy); in TensorHandleCopyBetweenDevices()
265 TFE_DeleteTensorHandle(hcpu); in TensorHandleCopyBetweenDevices()
303 TFE_DeleteTensorHandle(hcopy); in TensorHandleCopyBetweenDevicesError()
304 TFE_DeleteTensorHandle(hcpu); in TensorHandleCopyBetweenDevicesError()
[all …]
Dcustom_device_test.cc59 TFE_DeleteTensorHandle(retval); in TEST()
60 TFE_DeleteTensorHandle(hcpu); in TEST()
61 TFE_DeleteTensorHandle(hdevice); in TEST()
129 [var_handle]() { TFE_DeleteTensorHandle(var_handle); }); in TEST()
132 std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> one( in TEST()
133 TestScalarTensorHandle(context.get(), 111.f), TFE_DeleteTensorHandle); in TEST()
159 [var_value]() { TFE_DeleteTensorHandle(var_value); }); in TEST()
214 [var_handle]() { TFE_DeleteTensorHandle(var_handle); }); in TEST()
217 std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> one( in TEST()
218 TestScalarTensorHandle(context.get(), 111.f), TFE_DeleteTensorHandle); in TEST()
[all …]
Dc_api_remote_test_util.cc196 TFE_DeleteTensorHandle(retval_task0); in TestRemoteExecuteSilentCopies()
206 TFE_DeleteTensorHandle(h0_task0); in TestRemoteExecuteSilentCopies()
207 TFE_DeleteTensorHandle(h1_task0); in TestRemoteExecuteSilentCopies()
209 TFE_DeleteTensorHandle(packed_handle); in TestRemoteExecuteSilentCopies()
211 TFE_DeleteTensorHandle(h1_task2); in TestRemoteExecuteSilentCopies()
212 TFE_DeleteTensorHandle(retvals[0]); in TestRemoteExecuteSilentCopies()
214 TFE_DeleteTensorHandle(h); in TestRemoteExecuteSilentCopies()
217 TFE_DeleteTensorHandle(h); in TestRemoteExecuteSilentCopies()
Dc_api_distributed_test.cc211 TFE_DeleteTensorHandle(packed_handle); in TestFunctionWithPackedInput()
214 TFE_DeleteTensorHandle(retvals[0]); in TestFunctionWithPackedInput()
221 TFE_DeleteTensorHandle(h0); in TestFunctionWithPackedInput()
222 TFE_DeleteTensorHandle(h1); in TestFunctionWithPackedInput()
223 TFE_DeleteTensorHandle(h2); in TestFunctionWithPackedInput()
398 TFE_DeleteTensorHandle(retvals[0]); in TEST()
406 TFE_DeleteTensorHandle(var_handle); in TEST()
526 TFE_DeleteTensorHandle(retvals[0]); in TestDistributedFunctionCancellation()
535 TFE_DeleteTensorHandle(var_handle); in TestDistributedFunctionCancellation()
603 TFE_DeleteTensorHandle(h0_task0); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
[all …]
Dc_api_remote_test.cc97 TFE_DeleteTensorHandle(h0_task0); in TestRemoteExecute()
98 TFE_DeleteTensorHandle(h1_task0); in TestRemoteExecute()
99 TFE_DeleteTensorHandle(h0_task1); in TestRemoteExecute()
100 TFE_DeleteTensorHandle(h1_task1); in TestRemoteExecute()
101 TFE_DeleteTensorHandle(retvals[0]); in TestRemoteExecute()
Dc_api_cluster_test.cc78 TFE_DeleteTensorHandle(retval_task0); in CheckRemoteMatMulExecutesOK()
79 TFE_DeleteTensorHandle(h0_task0); in CheckRemoteMatMulExecutesOK()
80 TFE_DeleteTensorHandle(retvals[0]); in CheckRemoteMatMulExecutesOK()
176 TFE_DeleteTensorHandle(h0_task0_new); in TestRemoteExecuteChangeServerDef()
177 TFE_DeleteTensorHandle(h0_task1_new); in TestRemoteExecuteChangeServerDef()
282 TFE_DeleteTensorHandle(value_handle); in TestRemoteExecuteUpdateServerDefResourceAccess()
327 TFE_DeleteTensorHandle(var_handle0); in TestRemoteExecuteUpdateServerDefResourceAccess()
328 TFE_DeleteTensorHandle(var_handle1); in TestRemoteExecuteUpdateServerDefResourceAccess()
465 TFE_DeleteTensorHandle(var_handle0); in TestConnectToCluster()
466 TFE_DeleteTensorHandle(var_handle1); in TestConnectToCluster()
[all …]
Dc_api_experimental_test.cc308 TFE_DeleteTensorHandle(h); in TEST()
310 TFE_DeleteTensorHandle(result[0]); in TEST()
338 TFE_DeleteTensorHandle(m); in Executor_MatMul_CPU()
343 TFE_DeleteTensorHandle(retvals[0]); in Executor_MatMul_CPU()
365 TFE_DeleteTensorHandle(static_cast<TFE_TensorHandle*>(tensor_handle)); in Deleter()
404 TFE_DeleteTensorHandle(copy_aliased); // Note that this will delete copy. in TEST()
405 TFE_DeleteTensorHandle(on_host); in TEST()
409 TFE_DeleteTensorHandle(m); in TEST()
472 TFE_DeleteTensorHandle(retvals[0]); in TEST()
473 TFE_DeleteTensorHandle(hgpu); in TEST()
[all …]
Dc_api_debug_test.cc37 TFE_DeleteTensorHandle(h); in TEST()
59 TFE_DeleteTensorHandle(h);
Dcustom_device_testutil.cc45 ~LoggedTensor() { TFE_DeleteTensorHandle(tensor); } in ~LoggedTensor()
Dc_api_test_util.cc228 std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> in TestVariable()
230 TFE_DeleteTensorHandle); in TestVariable()
Dc_api.cc274 void TFE_DeleteTensorHandle(TFE_TensorHandle* h) { in TFE_DeleteTensorHandle() function
430 TFE_DeleteTensorHandle(result_handle); in CopyTensorToDevice()
447 TFE_DeleteTensorHandle(result_handle); in CopyTensorFromDevice()
462 TFE_DeleteTensorHandle(outputs[i]); in Execute()
Dc_api.h143 TF_CAPI_EXPORT extern void TFE_DeleteTensorHandle(TFE_TensorHandle* h);
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dsaved_model_api_test.cc144 TFE_DeleteTensorHandle(compute_fn_outputs[0]); in TEST_P()
145 TFE_DeleteTensorHandle(input_a); in TEST_P()
146 TFE_DeleteTensorHandle(input_b); in TEST_P()
284 TFE_DeleteTensorHandle(compute_fn_outputs[0]); in TEST_P()
285 TFE_DeleteTensorHandle(input_a); in TEST_P()
286 TFE_DeleteTensorHandle(input_b); in TEST_P()
341 TFE_DeleteTensorHandle(read_file_fn_outputs[0]); in TEST_P()
409 TFE_DeleteTensorHandle(input_foo); in TEST_P()
410 TFE_DeleteTensorHandle(lookup_fn_outputs[0]); in TEST_P()
441 TFE_DeleteTensorHandle(input_foo); in TEST_P()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Deval_util.cc116 MakeCleanup([input_handle] { TFE_DeleteTensorHandle(input_handle); }); in EvaluateOperation()
128 TFE_DeleteTensorHandle(tensor_handle); in EvaluateOperation()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor_test_util.cc36 TFE_DeleteTensorHandle(handle); in PYBIND11_MODULE()
Dpywrap_tensor_conversion.h94 TFE_DeleteTensorHandle(p.second); in DecrefUnrefAll()
Dpywrap_tensor.cc255 TFE_DeleteTensorHandle(output); in EagerCast()
495 TFE_DeleteTensorHandle(self->handle); in EagerTensor_dealloc()
/external/tensorflow/tensorflow/python/lib/core/
Dsafe_ptr.h35 void operator()(TFE_TensorHandle* p) const { TFE_DeleteTensorHandle(p); } in operator()
/external/tensorflow/tensorflow/c/experimental/saved_model/public/
DREADME.md22 TFE_DeleteTensorHandle(output);
/external/tensorflow/tensorflow/cc/experimental/base/public/
Dtensorhandle.h69 void operator()(TFE_TensorHandle* p) const { TFE_DeleteTensorHandle(p); } in operator()
/external/tensorflow/tensorflow/go/
Dtensor_handle.go59 C.TFE_DeleteTensorHandle(th.c)
/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_jni.cc59 TFE_DeleteTensorHandle(reinterpret_cast<TFE_TensorHandle*>(handle)); in Java_org_tensorflow_EagerOperation_deleteTensorHandle()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib.h42 TFE_DeleteTensorHandle(to_delete); in operator()
Dparallel_device_testlib.cc75 TFE_DeleteTensorHandle(handle_); in Destroy()
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc232 TFE_DeleteTensorHandle(scalar); in TEST_F()