Home
last modified time | relevance | path

Searched refs:AsAsync (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_executor.h71 virtual AsyncEagerNode* AsAsync() { return nullptr; } in AsAsync() function
87 AsyncEagerNode* AsAsync() final { return this; } in AsAsync() function
Deager_executor.cc104 if (node->AsAsync() != nullptr) { in SyncExecute()
221 bool async = item->node->AsAsync() != nullptr; in NodeDone()
383 AsyncEagerNode* async_node = item->node->AsAsync(); in RunItem()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.h34 AsyncOpKernel* AsAsync() override;
Dtpu_execute_op.cc596 AsyncOpKernel* TPUExecuteOp::AsAsync() { in AsAsync() function in tensorflow::TPUExecuteOp
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.h242 AsyncOpKernel* AsAsync() override;
Diterator_ops.cc952 AsyncOpKernel* IteratorGetNextOp::AsAsync() { in AsAsync() function in tensorflow::data::IteratorGetNextOp
/external/tensorflow/tensorflow/core/common_runtime/
Dimmutable_executor_state.cc139 item->kernel_is_async = (item->kernel->AsAsync() != nullptr); in Initialize()
Dexecutor.cc596 AsyncOpKernel* async_kernel = item.kernel->AsAsync(); in ProcessAsync()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h136 virtual AsyncOpKernel* AsAsync() { return nullptr; } in AsAsync() function
225 AsyncOpKernel* AsAsync() override { return this; } in AsAsync() function
Dop_kernel.cc1770 CHECK_EQ(nullptr, ctx->params_->op_kernel->AsAsync()) in CheckNotInComputeAsync()