Searched refs:ImmediateExecutionOperation (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_operation.h | 40 class ImmediateExecutionOperation : public AbstractOperation {
77 explicit ImmediateExecutionOperation(AbstractOperationKind kind) in ImmediateExecutionOperation() function
79 ~ImmediateExecutionOperation() override {} in ~ImmediateExecutionOperation()
84 void operator()(ImmediateExecutionOperation* p) const { in operator()
93 std::unique_ptr<ImmediateExecutionOperation,
|
| D | tfe_op_internal.h | 31 DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionOperation, TFE_Op);
32 DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionOperation*, TFE_Op*);
|
| D | immediate_execution_context.h | 99 ImmediateExecutionOperation* CreateOperation() override = 0;
|
| D | c_api.cc | 451 tensorflow::Status Execute(const ImmediateExecutionOperation* op, in Execute()
638 tensorflow::ImmediateExecutionOperation* new_op = in TFE_NewOp()
895 tensorflow::ImmediateExecutionOperation* unwrapped_op = in TFE_Execute()
|
| D | c_api_experimental.cc | 41 tensorflow::ImmediateExecutionOperation* op = in TFE_OpReset()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device_op_handler.h | 36 Status Execute(ImmediateExecutionOperation* op,
42 const ImmediateExecutionOperation& op) const;
|
| D | eager_operation.h | 37 class EagerOperation : public ImmediateExecutionOperation {
40 : ImmediateExecutionOperation(kEager), ctx_(*ctx) {} in EagerOperation()
275 ImmediateExecutionOperation* operation) { in OperationFromInterface()
280 const ImmediateExecutionOperation* operation) { in OperationFromInterface()
|
| D | custom_device_op_handler.cc | 54 Status CustomDeviceOpHandler::Execute(ImmediateExecutionOperation* op, in Execute()
99 CustomDevice** device, const ImmediateExecutionOperation& op) const { in MaybePinToCustomDevice()
|
| D | custom_device.h | 46 virtual Status Execute(const ImmediateExecutionOperation* op,
|
| D | core.cc | 183 ImmediateExecutionOperation* EagerContext::CreateOperation() { in CreateOperation()
|
| D | custom_device_test.cc | 49 Status Execute(const ImmediateExecutionOperation* op, in Execute()
|
| D | context.h | 133 ImmediateExecutionOperation* CreateOperation() override;
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 614 tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op); in TFE_InferShapes()
|