Searched refs:EagerOperation (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_operation.cc | 18 tensorflow::Status EagerOperation::SetDevice(const char* device) { in SetDevice()
28 void EagerOperation::AddInput(tensorflow::TensorHandle* h) { in AddInput()
34 void EagerOperation::ConsumeInput(tensorflow::TensorHandle* h) { in ConsumeInput()
39 string EagerOperation::DebugString() const { in DebugString()
|
| D | eager_operation.h | 23 class EagerOperation {
25 EagerOperation(tensorflow::EagerContext* ctx, const char* op, in EagerOperation() function
34 ~EagerOperation() { in ~EagerOperation()
|
| D | execute.cc | 91 Status MaybeCopyInputToExpectedDevice(EagerOperation* op, in MaybeCopyInputToExpectedDevice()
182 EagerOperation* op, in ValidateInputTypeAndPlacement()
225 Status GetOutputDTypes(EagerOperation* op, DataTypeVector* output_dtypes) { in GetOutputDTypes()
321 const EagerOperation* op, in AddInputDevicesToCacheKey()
366 Status EagerLocalExecute(EagerOperation* op, in EagerLocalExecute()
640 Status EagerRemoteExecute(EagerOperation* op, TensorHandle** retvals, in EagerRemoteExecute()
810 Status MaybeUpdateOpDevice(EagerOperation* op) { in MaybeUpdateOpDevice()
893 Status EagerExecute(EagerOperation* op, in EagerExecute()
1062 tensorflow::EagerOperation op(ctx, "_Send", /*is_function=*/false, types); in ExecuteSend()
1093 tensorflow::EagerOperation op(ctx, "_Recv", /*is_function=*/false, types); in ExecuteRecv()
|
| D | execute.h | 40 EagerOperation* op,
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 201 tensorflow::EagerOperation* op() { return op_.get(); } in op()
250 op_.reset(new tensorflow::EagerOperation(eager_context, name_.c_str(), in BuildEagerOp()
329 std::unique_ptr<tensorflow::EagerOperation> op_;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 154 std::unique_ptr<tensorflow::EagerOperation> op; in ExecuteOp()
168 op.reset(new tensorflow::EagerOperation(server_context->Context(), name, in ExecuteOp()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_internal.h | 118 tensorflow::EagerOperation operation;
|