Searched refs:compute_flatten_shape (Results 1 – 6 of 6) sorted by relevance
54 … input->info()->clone()->set_tensor_shape(misc::shape_calculator::compute_flatten_shape(input->inf… in configure()65 …fo_output = input->clone()->set_tensor_shape(misc::shape_calculator::compute_flatten_shape(input)); in validate()
62 … input->info()->clone()->set_tensor_shape(misc::shape_calculator::compute_flatten_shape(input->inf… in configure()73 …fo_output = input->clone()->set_tensor_shape(misc::shape_calculator::compute_flatten_shape(input)); in validate()
58 shape_flatten = compute_flatten_shape(&input_info); in setup()
208 …>clone()->set_is_resizable(true).reset_padding().set_tensor_shape(compute_flatten_shape(src)).set_… in configure_conv_fc()342 …>clone()->set_is_resizable(true).reset_padding().set_tensor_shape(compute_flatten_shape(src)).set_… in validate()
219 auto_init_if_empty(_flattened_src, src->clone()->set_tensor_shape(compute_flatten_shape(src))); in configure_conv_fc()383 …src->clone()->set_is_resizable(true).reset_padding().set_tensor_shape(compute_flatten_shape(src))); in validate()
561 inline TensorShape compute_flatten_shape(const ITensorInfo *input) in compute_flatten_shape() function