|
blace.ai
|
#include <graph_evaluator.h>
Public Member Functions | |
| GraphEvaluator (blace::ops::OpP node, bool is_in_worker=false) | |
| void | to_dot_file (const std::filesystem::path &filename) |
| std::string | to_dot_string () |
| std::pair< ml_core::ReturnCode, std::string > | evaluateToString () |
| std::pair< ml_core::ReturnCode, std::vector< int64_t > > | evaluateToIntList () |
| std::pair< ml_core::ReturnCode, std::shared_ptr< RawMemoryObject > > | evaluateToRawMemory (std::shared_ptr< ml_core::ProgressCallback > progress_callback=nullptr) |
A class to evaluate constructed graphs.
| blace::computation_graph::GraphEvaluator::GraphEvaluator | ( | blace::ops::OpP | node, |
| bool | is_in_worker = false |
||
| ) |
Constructs a graph evaluator instance from a node.
| node | The final node which shall be evaluated. |
| is_in_worker | If it is inside an IPC worker. |
| std::pair< ml_core::ReturnCode, std::vector< int64_t > > blace::computation_graph::GraphEvaluator::evaluateToIntList | ( | ) |
Evaluates graph to an integer list.
| std::pair< ml_core::ReturnCode, std::shared_ptr< RawMemoryObject > > blace::computation_graph::GraphEvaluator::evaluateToRawMemory | ( | std::shared_ptr< ml_core::ProgressCallback > | progress_callback = nullptr | ) |
Evaluates graph to at::Tensor.
| std::pair< ml_core::ReturnCode, std::string > blace::computation_graph::GraphEvaluator::evaluateToString | ( | ) |
Evaluates graph to string.
| void blace::computation_graph::GraphEvaluator::to_dot_file | ( | const std::filesystem::path & | filename | ) |
Writes a .dot file with the graph structure.
| filename |
| std::string blace::computation_graph::GraphEvaluator::to_dot_string | ( | ) |
Returns a .dot string with the graph structure.