Searched refs:TensorConstantMap (Results 1 – 3 of 3) sorted by relevance
14 using TensorConstantMap = std::unordered_map<std::string, at::Tensor*>; variable34 void update_inactive_constant_buffer(const TensorConstantMap& const_map);36 const TensorConstantMap& const_map,
161 const TensorConstantMap& const_map, in update_constant_buffer()172 const TensorConstantMap& const_map) { in update_inactive_constant_buffer()
108 torch::inductor::TensorConstantMap missing_map, rand_map, real_map; in test_aoti_constants_update()197 torch::inductor::TensorConstantMap rand_map, real_map; in test_aoti_double_buffering()271 torch::inductor::TensorConstantMap real_map; in test_aoti_double_buffering_with_tensor_constants()