#include "bncore/inference/workspace.hpp"
#include <algorithm>
#include <cassert>
#include <cstring>
#include <numeric>
#include <stdexcept>
Go to the source code of this file.
|
| static std::vector< std::size_t > | bncore::make_strides (const std::vector< std::size_t > &shape) |
| |
| static std::vector< NodeId > | bncore::combine_scopes (const std::vector< NodeId > &a, const std::vector< NodeId > &b, std::vector< int > &b_in_c) |
| |
| static std::size_t | bncore::build_pot_map_u32 (const std::vector< NodeId > &prod_scope, const std::vector< std::size_t > &prod_shape, const std::vector< NodeId > &src_scope, const std::vector< std::size_t > &src_shape, std::vector< uint32_t > &pot_map) |
| |
| static void | bncore::build_marg_map_u32 (const std::vector< NodeId > &src_scope, const std::vector< std::size_t > &src_shape, const std::vector< NodeId > &dst_scope, const std::vector< std::size_t > &dst_shape, std::vector< uint32_t > &marg_map) |
| |