Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/lib/gtl/
Dcleanup_test.cc37 auto s_cleaner = gtl::MakeCleanup([&s] { s.assign("cleaned"); }); in TEST()
49 auto s_cleaner = gtl::MakeCleanup([] { s.append(" clean"); }); in TEST()
58 auto s_cleaner = gtl::MakeCleanup([&s] { s.assign("cleaned"); }); in TEST()
84 auto c = gtl::MakeCleanup(Appender(&s, " cleaned")); in TEST()
94 auto clean1 = gtl::MakeCleanup(Appender(&s, " 1")); in TEST()
95 auto clean2 = gtl::MakeCleanup(Appender(&s, " 2")); in TEST()
107 auto clean1 = gtl::MakeCleanup(Appender(&s, " 1")); in TEST()
108 AnyCleanup clean2 = gtl::MakeCleanup(Appender(&s, " 2")); in TEST()
119 auto clean1 = gtl::MakeCleanup(Appender(&s, " 1")); in TEST()
120 auto clean2 = gtl::MakeCleanup(Appender(&s, " 2")); in TEST()
[all …]
Dcleanup.h106 TF_MUST_USE_RESULT Cleanup<DecayF> MakeCleanup(F&& f) { in MakeCleanup() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Deval_util.cc39 using gtl::MakeCleanup;
95 auto clean_status = MakeCleanup([status] { TF_DeleteStatus(status); }); in EvaluateOperation()
111 auto clean_op = MakeCleanup([op] { TFE_DeleteOp(op); }); in EvaluateOperation()
128 MakeCleanup([tf_tensor] { TF_DeleteTensor(tf_tensor); }); in EvaluateOperation()
132 MakeCleanup([input_handle] { TFE_DeleteTensorHandle(input_handle); }); in EvaluateOperation()
142 auto clean_outputs = MakeCleanup([&outputs] { in EvaluateOperation()
154 MakeCleanup([tf_tensor] { TF_DeleteTensor(tf_tensor); }); in EvaluateOperation()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.cc54 auto cleanup = tensorflow::gtl::MakeCleanup( in TransferLiteralFromDevice()
79 auto cleanup = tensorflow::gtl::MakeCleanup( in TransferLiteralFromDevice()
104 auto cleanup = tensorflow::gtl::MakeCleanup( in TransferLiteralToDevice()
119 auto cleanup = tensorflow::gtl::MakeCleanup( in TransferArrayFromDevice()
147 auto cleanup = tensorflow::gtl::MakeCleanup( in TransferArrayToDevice()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dasm_compiler.cc182 auto ptx_cleaner = tensorflow::gtl::MakeCleanup([&ptx_path] {
195 auto cubin_cleaner = tensorflow::gtl::MakeCleanup([&cubin_path] {
/external/tensorflow/tensorflow/core/framework/
Dmodel_test.cc43 auto cleanup_meta = gtl::MakeCleanup([async_interleave_many, meta_source]() { in TEST_P()
49 auto cleanup1 = gtl::MakeCleanup([async_interleave_many, source1]() { in TEST_P()
55 auto cleanup2 = gtl::MakeCleanup([async_interleave_many, source2]() { in TEST_P()
501 auto cleanup_meta = gtl::MakeCleanup([async_interleave_many, meta_source]() { in TEST()
511 auto cleanup1 = gtl::MakeCleanup([async_interleave_many, source1]() { in TEST()
517 auto cleanup2 = gtl::MakeCleanup([async_interleave_many, source2]() { in TEST()
Dmodel.cc153 auto cleanup = gtl::MakeCleanup( in OutputTimeLocked()
251 gtl::MakeCleanup([input_times]() { input_times->pop_back(); }); in OutputTimeLocked()
352 auto cleanup = gtl::MakeCleanup([input_times, old_input_time]() { in OutputTimeLocked()
471 gtl::MakeCleanup([input_times]() { input_times->pop_back(); }); in OutputTimeLocked()
558 auto cleanup = gtl::MakeCleanup([input_times, old_input_time]() { in OutputTimeLocked()
/external/tensorflow/tensorflow/c/eager/
Dtape.h666 auto cleanup = gtl::MakeCleanup([this, &state]() { in ComputeGradient()
913 gtl::MakeCleanup([&call_state] { call_state.backward_tape = nullptr; }); in ForwardpropFromTape()
943 auto delete_grad = gtl::MakeCleanup([&grad, this] { in ForwardpropFromTape()
1026 auto delete_new_zeros = gtl::MakeCleanup([&new_zeros, this] { in Accumulate()
1055 auto pop_call_state = gtl::MakeCleanup([this] { this->call_state_.pop(); }); in Accumulate()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dremote_device.cc91 auto cleanup = gtl::MakeCleanup( in NewRemoteDevices()
Dmaster_session.cc1308 auto cleanup = gtl::MakeCleanup([&workers, worker_cache] { in CreateWorkerSessions()
1406 auto cleanup = gtl::MakeCleanup([&workers, worker_cache] { in DeleteWorkerSessions()
1693 auto cleanup = gtl::MakeCleanup([this] { MarkRunCompletion(); }); in DoPartialRun()
1928 auto cleanup = gtl::MakeCleanup([this] { MarkRunCompletion(); }); in DoRunWithLocalExecution()
2018 auto cleanup = gtl::MakeCleanup([this] { MarkRunCompletion(); }); in DoRunCallable()
Dcluster_function_library_runtime.cc297 gtl::MakeCleanup([call_options, req, resp, local_status, done] { in Run()
/external/tensorflow/tensorflow/core/common_runtime/
Dkernel_benchmark_testlib.cc50 auto cleanup = gtl::MakeCleanup([g, init]() { in Benchmark()
/external/tensorflow/tensorflow/core/lib/gif/
Dgif_io.cc63 const auto cleanup = gtl::MakeCleanup([gif_file]() { in Decode()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc94 auto cleanup_handle = gtl::MakeCleanup([&]() { in FunctionalizeControlFlowForFunction()
Dresource_util.cc208 auto release_handle_on_return = gtl::MakeCleanup( in PropagateThroughCallOp()
Dgraph_compiler.cc115 auto output_registry_cleanup = gtl::MakeCleanup([&output_registry] { in Compile()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc91 auto cleanup = gtl::MakeCleanup(std::move(deregister_fn)); in GetNext()
152 auto cleanup = gtl::MakeCleanup(std::move(deregister_fn)); in Restore()
187 auto cleanup = gtl::MakeCleanup(std::move(deregister_fn)); in SetIteratorFromDataset()
Dprefetch_dataset_op.cc380 auto cleanup = gtl::MakeCleanup([this, ctx] { RecordStop(ctx.get()); }); in PrefetchThread()
/external/tensorflow/tensorflow/core/platform/cloud/
Dram_file_block_cache.cc103 gtl::MakeCleanup([this, &downloaded_block, &key, &block] { in MaybeFetch()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_executor.cc300 gtl::MakeCleanup([this] { thread_exited_notification_.Notify(); }); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.cc115 auto cleanup = tensorflow::gtl::MakeCleanup([&buffers]() { in TransferLiteralToInfeed()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_reduce_thunk.cc265 auto cuda_device_restorer = MakeCleanup( in Init()
Dgpu_executable.cc145 auto cleanup = MakeCleanup( in ExecuteThunks()
/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util.cc279 auto release_handle_on_return = gtl::MakeCleanup( in IsCompilableCall()
Dxla_cluster_util.cc460 auto release_handle_on_return = gtl::MakeCleanup( in DoesAnyCalleeHaveRefNodes()

12