Searched defs:set_tensor_map (Results 1 – 2 of 2) sorted by relevance
36 void set_tensor_map(const std::vector<int64_t> &tensor_map) { tensor_map_ = tensor_map; } in set_tensor_map() function
70 void set_tensor_map(const std::map<Tensor *, std::string> &tensor_map) { in set_tensor_map() function