Home
last modified time | relevance | path

Searched defs:cancellation_manager (Results 1 – 25 of 50) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dpartial_run_mgr_test.cc28 CancellationManager* cancellation_manager; in TEST() local
37 CancellationManager* cancellation_manager; in TEST() local
49 CancellationManager* cancellation_manager; in TEST() local
63 CancellationManager* cancellation_manager; in TEST() local
111 CancellationManager* cancellation_manager; in TEST_P() local
127 CancellationManager* cancellation_manager; in TEST_P() local
H A Dgraph_mgr.cc423 CancellationManager* cancellation_manager, in ExecuteAsync()
519 CostGraphDef* cost_graph, CancellationManager* cancellation_manager, in StartParallelExecutors()
H A Dpartial_run_mgr.h72 std::unique_ptr<CancellationManager> cancellation_manager; member
H A Dpartial_run_mgr.cc23 CancellationManager** cancellation_manager) { in FindOrCreate()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device.cc259 CancellationManager cancellation_manager; member
266 CancellationManager* cancellation_manager, in Run()
354 CancellationManager* cancellation_manager, in PrepareForRun()
415 CancellationManager* cancellation_manager, in Run()
460 CancellationManager* cancellation_manager, in RunAsync()
H A Dexecute_node.h99 CancellationManager* cancellation_manager, in ExecuteNode()
155 CancellationManager* cancellation_manager, in AsyncExecuteNode()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dmodel_test.cc1307 CancellationManager cancellation_manager; in TEST_P() local
2164 CancellationManager cancellation_manager; in TEST_F() local
2216 CancellationManager cancellation_manager; in TEST_F() local
2283 CancellationManager cancellation_manager; in TEST_F() local
2352 CancellationManager cancellation_manager; in TEST_F() local
2421 CancellationManager cancellation_manager; in TEST_F() local
H A Dmodel.cc2170 CancellationManager* cancellation_manager) { in Optimize()
2279 CancellationManager* cancellation_manager) { in OptimizeLoop()
2337 CancellationManager* cancellation_manager) { in OptimizeGradientDescent()
2404 CancellationManager* cancellation_manager, StopPredicate should_stop) { in OptimizeHillClimbHelper()
2508 CancellationManager* cancellation_manager) { in OptimizeStageBased()
2517 CancellationManager* cancellation_manager) { in OptimizeStageBasedParallelism()
2661 CancellationManager* cancellation_manager) { in OptimizeHillClimb()
2689 CancellationManager* cancellation_manager) { in OptimizeMaxParallelism()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_execute.cc323 OpKernelContext* ctx, CancellationManager* cancellation_manager, in RegisterCancellation()
363 OpKernelContext* ctx, CancellationManager* cancellation_manager, in UnregisterCancellation()
423 CancellationManager* cancellation_manager, OpKernelContext* ctx, in TPUExecute()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_test_util.cc63 CancellationManager* cancellation_manager, const StatusCallback& done) { in RecvFromPeer()
77 CancellationManager* cancellation_manager, in PostToPeer()
318 CancellationManager cancellation_manager; in RunCollective() local
H A Dbuf_rendezvous.cc90 CancellationManager* cancellation_manager) { in ProvideBuf()
152 CancellationManager* cancellation_manager) { in ConsumeBuf()
H A Dcollective_rma_local.cc31 CancellationManager* cancellation_manager, const StatusCallback& done) { in RecvFromPeer()
102 CancellationManager* cancellation_manager, const StatusCallback& done) { in PostToPeer()
H A Dbuf_rendezvous.h70 CancellationManager* cancellation_manager; member
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dfunction.py441 def call(self, ctx, args, cancellation_manager=None): argument
1576 def _call_impl(self, args, kwargs, cancellation_manager=None): argument
1594 def _call_with_flat_signature(self, args, kwargs, cancellation_manager): argument
1643 def _call_with_structured_signature(self, args, kwargs, cancellation_manager): argument
1756 def _call_flat(self, args, captured_inputs, cancellation_manager=None): argument
1851 def _experimental_with_cancellation_manager(self, cancellation_manager): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dcaptured_function.cc801 CancellationManager cancellation_manager(ctx->cancellation_manager()); in Run() local
864 CancellationManager cancellation_manager(ctx->cancellation_manager()); in RunWithBorrowedArgs() local
919 CancellationManager cancellation_manager; in RunInstantiated() local
966 auto cancellation_manager = in RunAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_experimental.cc551 TFE_CancellationManager* cancellation_manager) { in TFE_CancellationManagerStartCancel()
556 TFE_CancellationManager* cancellation_manager) { in TFE_CancellationManagerIsCancelled()
561 TFE_CancellationManager* cancellation_manager) { in TFE_DeleteCancellationManager()
566 TFE_CancellationManager* cancellation_manager, in TFE_OpSetCancellationManager()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device_lib.cc197 CancellationManager& cancellation_manager, in StartExecute()
351 CancellationManager& cancellation_manager, in StartExecute()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Ddelegate_data.h62 tensorflow::CancellationManager* cancellation_manager) { in SetCancellationManager()
H A Dkernel.cc394 tensorflow::CancellationManager* cancellation_manager; member
705 static auto* const cancellation_manager = new tensorflow::CancellationManager; in GetDefaultCancellationManager() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dreduce_dataset_op.cc69 CancellationManager cancellation_manager(ctx->cancellation_manager()); in DoCompute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_compat_request_state.h125 CancellationManager* cancellation_manager() const { in cancellation_manager() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dto_tf_record_op.cc84 CancellationManager cancellation_manager(ctx->cancellation_manager()); in DoCompute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconditional_accumulator_base.cc111 CancellationManager* cancellation_manager, CancellationToken token) { in Cancel()
H A Dconditional_accumulator_base.h96 CancellationManager* cancellation_manager; // not owned member
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Dremote_execute_node.h44 CancellationManager* cancellation_manager, in RemoteExecuteNode()

12