Searched refs:map_input_to_tensor_bufs_ (Results 1 – 4 of 4) sorted by relevance
38 auto got = map_input_to_tensor_bufs_.find(arg->name_hint()); in areBufsInMap()39 if (got == map_input_to_tensor_bufs_.end()) { in areBufsInMap()47 auto input_ = map_input_to_tensor_bufs_.find(buf->name_hint()); in getMultiDimBuf()48 if (input_ != map_input_to_tensor_bufs_.end()) { in getMultiDimBuf()56 auto input_ = map_input_to_tensor_bufs_.find(buf->name_hint()); in getInputName()57 if (input_ != map_input_to_tensor_bufs_.end()) { in getInputName()76 map_input_to_tensor_bufs_ = loop_options.get_buffer_mapping(); in visit()
377 return map_input_to_tensor_bufs_; in getBufferMap()385 map_input_to_tensor_bufs_.emplace(t_buf->name_hint(), v->buf()); in visit()391 map_input_to_tensor_bufs_.emplace(v->buf()->name_hint(), v->buf()); in visit()395 std::unordered_map<std::string, BufPtr> map_input_to_tensor_bufs_; variable
49 return map_input_to_tensor_bufs_; in getBufferMap()57 std::unordered_map<std::string, BufPtr> map_input_to_tensor_bufs_; variable
684 map_input_to_tensor_bufs_ = map; in set_buffer_mapping()688 return map_input_to_tensor_bufs_; in get_buffer_mapping()695 std::unordered_map<std::string, BufPtr> map_input_to_tensor_bufs_; variable