Searched defs:shape_flatten (Results 1 – 2 of 2) sorted by relevance
56 TensorShape shape_flatten; in setup() local76 …TensorType compute_target(const TensorShape &shape, const TensorShape &shape_flatten, DataType dat… in compute_target()105 …SimpleTensor<T> compute_reference(const TensorShape &shape, const TensorShape &shape_flatten, Data… in compute_reference()
35 SimpleTensor<T> flatten_layer(const SimpleTensor<T> &src, const TensorShape &shape_flatten) in flatten_layer()