Searched refs:compute_space_to_depth_shape (Results 1 – 3 of 3) sorted by relevance
81 …TensorShape output_shape = misc::shape_calculator::compute_space_to_depth_shape(input->info(), blo… in configure()
80 TensorShape output_shape = compute_space_to_depth_shape(input->info(), block_shape); in configure()
1177 inline TensorShape compute_space_to_depth_shape(const ITensorInfo *input, int32_t block_shape) in compute_space_to_depth_shape() function