Searched refs:wrap_with_logging_tensor (Results 1 – 1 of 1) sorted by relevance
38 def wrap_with_logging_tensor(ctor): function47 k: wrap_with_logging_tensor(ctor) for (k, ctor) in base_ctors_dict.items()