Skip to content

Commit

Permalink
rocAL - Parameter changes (ROCm#109)
Browse files Browse the repository at this point in the history
* Adding parameterVX changes

* Resolving review comments

* Adding API comments

* ResolvePR comments

* Resolve PR Comments

* Resolving review comments

* Fix name changes of the variable

* Resolving review comments

---------

Co-authored-by: Swetha B S <[email protected]>
  • Loading branch information
2 people authored and SundarRajan28 committed Jul 31, 2024
1 parent 9d829fa commit b423722
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions rocAL/include/pipeline/master_graph.h
Original file line number Diff line number Diff line change
Expand Up @@ -403,6 +403,7 @@ inline std::shared_ptr<Cifar10LoaderNode> MasterGraph::add_node(const std::vecto
return node;
}

#ifdef ROCAL_VIDEO
/*
* Explicit specialization for VideoLoaderNode
*/
Expand Down Expand Up @@ -445,6 +446,7 @@ inline std::shared_ptr<VideoLoaderSingleShardNode> MasterGraph::add_node(const s

return node;
}
#endif

template <>
inline std::shared_ptr<NumpyLoaderNode> MasterGraph::add_node(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs) {
Expand Down

0 comments on commit b423722

Please sign in to comment.