Searched defs:tensor_pool_fn (Results 1 – 2 of 2) sorted by relevance
502 def _tensor_pool_adjusted_model(model, tensor_pool_fn): argument575 tensor_pool_fn=None, argument
700 tensor_pool_fn = lambda x: random_tensor_pool.tensor_pool(x, pool_size=5) function719 def tensor_pool_fn(_): function