Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.cc37 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in Create()
38 TFE_NewOp(context, "VarHandleOp", status), TFE_DeleteOp); in Create()
62 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in Destroy()
63 TFE_NewOp(context, "DestroyResourceOp", status), &TFE_DeleteOp); in Destroy()
79 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in Read()
80 TFE_NewOp(context, "ReadVariableOp", status), &TFE_DeleteOp); in Read()
98 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in GeneralAssignment()
99 TFE_NewOp(context, op_name, status), &TFE_DeleteOp); in GeneralAssignment()
162 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in ExtractPerDeviceValues()
163 TFE_NewOp(context, "TPUReplicatedOutput", status), TFE_DeleteOp); in ExtractPerDeviceValues()
[all …]
Dparallel_device_lib_test.cc55 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op( in TEST()
56 TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); in TEST()
73 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> read_op( in TEST()
74 TFE_NewOp(context.get(), "ReadVariableOp", status.get()), TFE_DeleteOp); in TEST()
111 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op( in TEST()
112 TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); in TEST()
183 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> call_op( in TEST()
185 TFE_DeleteOp); in TEST()
275 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> size_op( in TEST()
276 TFE_NewOp(context.get(), "Size", status.get()), TFE_DeleteOp); in TEST()
Dparallel_device_testlib.h121 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in CreatePerDeviceValues()
122 TFE_NewOp(context, "TPUReplicatedInput", status), TFE_DeleteOp); in CreatePerDeviceValues()
Dparallel_device_lib.h240 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> const_op( in ScalarsFromSequence()
241 TFE_NewOp(context, "Const", status), TFE_DeleteOp); in ScalarsFromSequence()
Dparallel_device_test.cc365 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in TEST()
367 TFE_DeleteOp); in TEST()
385 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in CollectiveSum()
386 TFE_NewOp(context, "CollectiveReduce", status), TFE_DeleteOp); in CollectiveSum()
550 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in TEST()
551 TFE_NewOp(context.get(), function_name, status.get()), TFE_DeleteOp); in TEST()
Dparallel_device.cc36 void operator()(TFE_Op* to_delete) const { TFE_DeleteOp(to_delete); } in operator ()()
Dparallel_device_lib.cc32 void operator()(TFE_Op* to_delete) const { TFE_DeleteOp(to_delete); } in operator ()()
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc64 TFE_DeleteOp(matmul); in BM_InitOp()
107 TFE_DeleteOp(matmul); in BM_Execute()
146 TFE_DeleteOp(identity); in BM_Execute_Identity()
448 TFE_DeleteOp(matmul); in TensorHandleSilentCopy()
502 TFE_DeleteOp(matmul); in SetAndGetOpDevices()
593 TFE_DeleteOp(shape_op); in TEST()
651 TFE_DeleteOp(add_op); in ExecuteAdd()
734 TFE_DeleteOp(matmul); in Execute_MatMul_CPU()
776 TFE_DeleteOp(matmul); in Execute_MatMul_CPU_Runtime_Error()
813 TFE_DeleteOp(matmul2); in Execute_MatMul_CPU_Runtime_Error()
[all …]
Dcustom_device_test.cc49 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> matmul( in TEST()
50 MatMulOp(context, hcpu, hdevice), TFE_DeleteOp); in TEST()
82 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> reused_op( in TEST()
83 TFE_NewOp(context.get(), "Identity", status.get()), TFE_DeleteOp); in TEST()
113 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in TEST()
114 TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); in TEST()
198 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in TEST()
199 TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); in TEST()
298 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> matmul( in TEST()
299 MatMulOp(context.get(), hcpu.get(), hcpu.get()), TFE_DeleteOp); in TEST()
Dc_api_distributed_test.cc210 TFE_DeleteOp(func); in TestFunctionWithPackedInput()
405 TFE_DeleteOp(func); in TEST()
534 TFE_DeleteOp(func); in TestDistributedFunctionCancellation()
609 TFE_DeleteOp(matmul); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
Dc_api_cluster_test.cc82 TFE_DeleteOp(matmul); in CheckRemoteMatMulExecutesOK()
103 TFE_DeleteOp(op); in ReadVariable()
312 TFE_DeleteOp(op); in TestRemoteExecuteUpdateServerDefResourceAccess()
325 TFE_DeleteOp(op); in TestRemoteExecuteUpdateServerDefResourceAccess()
Dc_api_remote_test.cc103 TFE_DeleteOp(matmul); in TestRemoteExecute()
Dc_api_experimental_test.cc296 TFE_DeleteOp(op); in TEST()
337 TFE_DeleteOp(matmul); in Executor_MatMul_CPU()
471 TFE_DeleteOp(shape_op); in TEST()
Dc_api_remote_test_util.cc220 TFE_DeleteOp(matmul); in TestRemoteExecuteSilentCopies()
Dcustom_device_testutil.cc133 TFE_DeleteOp(op); in LoggingDeviceExecute()
Dc_api_test_util.cc213 TFE_DeleteOp(op); in TestVariable()
238 TFE_DeleteOp(op); in TestVariable()
Dc_api.h251 TF_CAPI_EXPORT extern void TFE_DeleteOp(TFE_Op* op);
Dc_api.cc648 void TFE_DeleteOp(TFE_Op* op) { in TFE_DeleteOp() function
719 TFE_DeleteOp(op); in TFE_OpNameGetAttrType()
1056 TFE_DeleteOp(func_op); in SetOpAttrValueScalar()
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dsaved_model_api_test.cc147 TFE_DeleteOp(compute_fn_op); in TEST_P()
287 TFE_DeleteOp(serving_default_op); in TEST_P()
342 TFE_DeleteOp(read_file_op); in TEST_P()
411 TFE_DeleteOp(lookup_op); in TEST_P()
443 TFE_DeleteOp(lookup_op); in TEST_P()
476 TFE_DeleteOp(lookup_op); in TEST_P()
576 TFE_DeleteOp(while_fn_op); in TEST_P()
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc194 TFE_DeleteOp(matmul_op); in TEST_F()
213 TFE_DeleteOp(reshape_op); in TEST_F()
229 TFE_DeleteOp(fill_op); in TEST_F()
/external/tensorflow/tensorflow/c/experimental/saved_model/public/
DREADME.md23 TFE_DeleteOp(op);
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Deval_util.cc89 auto clean_op = MakeCleanup([op] { TFE_DeleteOp(op); }); in EvaluateOperation()
/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_jni.cc53 TFE_DeleteOp(reinterpret_cast<TFE_Op*>(handle)); in Java_org_tensorflow_EagerOperation_delete()
Deager_operation_builder_jni.cc88 TFE_DeleteOp(reinterpret_cast<TFE_Op*>(op_handle)); in Java_org_tensorflow_EagerOperationBuilder_delete()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc239 TFE_DeleteOp(op); \ in EagerCast()
259 TFE_DeleteOp(op); in EagerCast()

12