Searched defs:shape_flatten (Results 1 – 4 of 4) sorted by relevance
56 TensorShape shape_flatten; in setup() local73 …TensorType compute_target(const TensorShape &shape, const TensorShape &shape_flatten, DataType dat… in compute_target()102 …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()
212 TensorShape shape_flatten = compute_flatten_shape(input->info()); in configure_conv_fc() local
217 TensorShape shape_flatten = compute_flatten_shape(input->info()); in configure_conv_fc() local