Searched refs:compute_unpool_shape (Results 1 – 3 of 3) sorted by relevance
110 const TensorShape output_shape = compute_unpool_shape(*src, pool_info); in configure()
102 const TensorShape output_shape = compute_unpool_shape(*input->info(), pool_info); in configure()
835 inline TensorShape compute_unpool_shape(const ITensorInfo &input, PoolingLayerInfo pool_info) in compute_unpool_shape() function