| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor.cc | 36 tpu::ExecutorApiFn()->TpuExecutor_FreeFn(executor_); in ~TpuExecutor()
45 tpu::ExecutorApiFn()->TpuExecutor_InitFn(executor_, device_ordinal, options, in Init()
52 return tpu::ExecutorApiFn()->TpuExecutor_PlatformDeviceCountFn(executor_); in PlatformDeviceCount()
56 tpu::ExecutorApiFn()->TpuExecutor_SyncAndForgetFailedStreamsFn(executor_); in SyncAndForgetFailedStreams()
60 return tpu::ExecutorApiFn()->TpuExecutor_SynchronizeAllActivityFn(executor_); in SynchronizeAllActivity()
66 executor_, get_stream(stream->implementation()), status.c_status); in BlockHostUntilDone()
72 tpu::ExecutorApiFn()->TpuExecutor_BlockUntilDoneOrFailedFn(executor_, in BlockUntilDoneOrFailed()
80 executor_, get_stream(stream->implementation()), status.c_status); in GetStatus()
86 tpu::ExecutorApiFn()->TpuExecutor_GetCoreLocationFn(executor_)); in GetCoreLocationExternal()
91 executor_, get_stream(stream->implementation())); in AllocateStream()
[all …]
|
| D | tpu_executor.h | 62 : platform_(platform), executor_(executor) {} in TpuExecutor()
220 SE_StreamExecutor* se_executor() { return executor_; } in se_executor()
238 SE_StreamExecutor* executor_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_op_rewrite_registry_test.cc | 25 : EagerOpRewrite(name, file, line), executor_(/*async=*/false) {} in TestEagerOpRewrite()
27 EagerExecutor executor_; member in tensorflow::TestEagerOpRewrite
34 TF_RETURN_IF_ERROR(op->Reset("NoOp", nullptr, false, &executor_)); in Run()
|
| D | mkl_eager_op_rewrite_test.cc | 48 EagerExecutor executor_(false); in CreateOp() local
52 op.get()->Reset(op_name.c_str(), nullptr, false, &executor_)); in CreateOp()
|
| D | eager_operation.h | 184 EagerExecutor& Executor() { return *executor_; } in Executor()
254 EagerExecutor* executor_; // Not owned. variable
|
| D | eager_operation.cc | 346 executor_ = executor ? executor : &ctx_.Executor(); in Reset()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 48 executor_ = backend_->default_stream_executor(); in BufferDonationTest()
49 TF_CHECK_OK(executor_->Init()); in BufferDonationTest()
56 se::StreamExecutor* executor_; member in xla::__anonf885129d0111::BufferDonationTest
70 std::move(hlo_module), executor_, in RunAndCheck()
76 backend_->compiler()->RunBackend(std::move(hlo_module), executor_, in RunAndCheck()
79 se::Stream stream(executor_); in RunAndCheck()
103 executor_->device_ordinal())); in RunAndCheck()
115 input_buffers.element(index), executor_->device_ordinal(), in RunAndCheck()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | local_device_state.h | 96 se::StreamExecutor* executor() const { return executor_; } in executor()
98 int device_ordinal() const { return executor_->device_ordinal(); } in device_ordinal()
174 se::StreamExecutor* const executor_; variable
|
| D | local_device_state.cc | 37 executor_(executor), in LocalDeviceState()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 758 : executor_(executor), module_handle_(module_handle) {} in ScopedModuleHandle()
761 executor_ = other.executor_; in ScopedModuleHandle()
763 other.executor_ = nullptr; in ScopedModuleHandle()
768 executor_ = other.executor_;
770 other.executor_ = nullptr;
777 CHECK(executor_->UnloadModule(module_handle_)); in ~ScopedModuleHandle()
782 StreamExecutor *executor_;
|
| /external/libchrome/base/ |
| D | deferred_sequenced_task_runner_unittest.cc | 29 : executor_(executor), in ExecuteTaskOnDestructor()
34 virtual ~ExecuteTaskOnDestructor() { executor_->ExecuteTask(task_id_); } in ~ExecuteTaskOnDestructor()
35 DeferredSequencedTaskRunnerTest* executor_; member in base::__anon03a06eda0111::DeferredSequencedTaskRunnerTest::ExecuteTaskOnDestructor
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 64 executor_ = [[::CoreMlExecutor alloc] init];
71 NSURL* model_url = [executor_ saveModel:model_.get()];
73 if (![executor_ build:model_url]) {
218 if (![executor_ invokeWithInputs:inputs_ outputs:outputs_]) {
232 CoreMlDelegateKernel::~CoreMlDelegateKernel() { [executor_ cleanup]; }
|
| D | coreml_delegate_kernel.h | 59 ::CoreMlExecutor* executor_;
|
| /external/webrtc/test/pc/e2e/ |
| D | peer_connection_quality_test.cc | 114 executor_(std::make_unique<TestActivitiesExecutor>( in PeerConnectionE2EQualityTest()
139 executor_->ScheduleActivity(target_time_since_start, absl::nullopt, func); in ExecuteAt()
146 executor_->ScheduleActivity(initial_delay_since_start, interval, func); in ExecuteEvery()
309 executor_->ScheduleActivity(TimeDelta::Zero(), kStatsUpdateInterval, in Run()
334 executor_->Start(task_queue_.get()); in Run()
348 executor_->Stop(); in Run()
|
| D | peer_connection_quality_test.h | 116 std::unique_ptr<TestActivitiesExecutor> executor_; variable
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.h | 165 EagerExecutor* const executor_; variable
|
| D | remote_copy_node.cc | 82 executor_(executor), in RemoteCopyNode()
100 TF_RETURN_IF_ERROR(executor_->status()); in RunLocalSend()
190 TF_RETURN_IF_ERROR(executor_->status()); in RunLocalRecv()
|
| /external/libtextclassifier/native/annotator/ |
| D | model-executor.h | 104 std::unique_ptr<TfLiteModelExecutor> executor_; variable
|
| D | model-executor.cc | 95 : executor_(std::move(executor)), in TFLiteEmbeddingExecutor()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 97 se::StreamExecutor* executor() const { return executor_; } in executor()
126 se::StreamExecutor* executor_; // not owned variable
|
| D | gpu_device.cc | 417 TF_RETURN_IF_ERROR(executor_->SynchronousMemZero( in InitScratchBuffers()
432 executor_ = executor_status.ValueOrDie(); in Init()
435 tf_gpu_id_, 0, executor_, options.config.gpu_options()); in Init()
444 em_ = EventMgrFactory::Singleton()->GetEventMgr(executor_, in Init()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 73 executor_(TFE_NewExecutor(is_async)), in DeviceThread()
140 ExecutorPtr executor_ TF_GUARDED_BY(execution_mutex_);
228 TFE_ContextSetExecutorForThread(context, executor_.get()); in Execute()
|
| /external/tensorflow/tensorflow/core/platform/s3/ |
| D | s3_file_system.h | 124 std::shared_ptr<Aws::Utils::Threading::PooledThreadExecutor> executor_; variable
|
| D | s3_file_system.cc | 456 executor_(nullptr, ShutdownExecutor) { in S3FileSystem()
562 if (this->executor_.get() == nullptr) { in GetExecutor()
563 this->executor_ = in GetExecutor()
567 return this->executor_; in GetExecutor()
|