Home
last modified time | relevance | path

Searched refs:OpHandlerOperation (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/op_handler/
Dinternal.cc47 OpHandlerOperation* OpHandlerContext::CreateOperation() { in CreateOperation()
48 OpHandlerOperation* result = in CreateOperation()
49 new OpHandlerOperation(parent_ctx_->CreateOperation()); in CreateOperation()
56 OpHandlerOperation::OpHandlerOperation(AbstractOperation* parent_op) in OpHandlerOperation() function in tensorflow::OpHandlerOperation
59 OpHandler* OpHandlerOperation::get_handler() { return handler_.get(); } in get_handler()
61 void OpHandlerOperation::set_handler(OpHandler* handler) { in set_handler()
68 Status OpHandlerOperation::Execute(absl::Span<AbstractTensorHandle*> retvals, in Execute()
Dinternal.h29 class OpHandlerOperation; variable
42 virtual Status Execute(OpHandlerOperation* operation,
66 class OpHandlerOperation : public WrapperOperation {
68 explicit OpHandlerOperation(AbstractOperation*);
91 OpHandlerOperation* CreateOperation() override;
109 void operator()(OpHandlerOperation* operation) { operation->Release(); } in operator()
112 typedef std::unique_ptr<OpHandlerOperation, ReleaseOpHandlerOperation>
Dinternal_test.cc29 Status Execute(OpHandlerOperation* operation, in Execute()