Tensorflow: How to call an operation in C++ from another operation
Tensorflow: How to call an operation in C++ from another operation
I am writing a new operation in C++ and would like to make use of another operation. For example, I construct two new tensors and would like to do matrix multiplication in my operation so I would like to make use of the MatMulOp
. However, the MatMulOp::Compute
requires a OpKernelContext
as input.
MatMulOp
MatMulOp::Compute
OpKernelContext
My question is how I can form such input to call this operation?
By clicking "Post Your Answer", you acknowledge that you have read our updated terms of service, privacy policy and cookie policy, and that your continued use of the website is subject to these policies.
Comments
Post a Comment