Searched refs:flatten_layer (Results 1 – 4 of 4) sorted by relevance
35 SimpleTensor<T> flatten_layer(const SimpleTensor<T> &src, const TensorShape &shape_flatten) in flatten_layer() function45 template SimpleTensor<float> flatten_layer(const SimpleTensor<float> &src, const TensorShape &shape…46 template SimpleTensor<half> flatten_layer(const SimpleTensor<half> &src, const TensorShape &shape_f…
39 SimpleTensor<T> flatten_layer(const SimpleTensor<T> &src, const TensorShape &shape_flatten);
83 FunctionType flatten_layer; in compute_target() local84 flatten_layer.configure(&src, &dst); in compute_target()100 flatten_layer.run(); in compute_target()113 return reference::flatten_layer<T>(src, shape_flatten); in compute_reference()
366 flatten_layer = core.Flatten()368 y = flatten_layer(x)