Searched defs:dtype_map (Results 1 – 3 of 3) sorted by relevance
676 def build_tensor(tensors, dtype_map): argument
133 const std::unordered_map<std::type_index, dtype_t> dtype_map = { variable
365 def __init__(self, save_path, dtype_map=None): argument