Searched refs:compute_space_to_depth_shape (Results 1 – 4 of 4) sorted by relevance
73 …auto calc_out_shape = misc::shape_calculator::compute_space_to_depth_shape(input.info(), block_sha… in compute_target()
81 …TensorShape output_shape = misc::shape_calculator::compute_space_to_depth_shape(input->info(), blo… in configure()
82 TensorShape output_shape = compute_space_to_depth_shape(input->info(), block_shape); in configure()
1191 inline TensorShape compute_space_to_depth_shape(const ITensorInfo *input, int32_t block_shape) in compute_space_to_depth_shape() function