Searched defs:flatten_layer (Results 1 – 2 of 2) sorted by relevance
35 SimpleTensor<T> flatten_layer(const SimpleTensor<T> &src, const TensorShape &shape_flatten) in flatten_layer() function
83 FunctionType flatten_layer; in compute_target() local