Searched refs:compute_batch_to_space_shape (Results 1 – 3 of 3) sorted by relevance
106 …TensorShape output_shape = compute_batch_to_space_shape(input->info(), block_shape_x, block_shape_… in configure()
131 …TensorShape output_shape = compute_batch_to_space_shape(input->info(), block_shape_x, block_shape_… in configure()
1081 inline TensorShape compute_batch_to_space_shape(const ITensorInfo *input, const int block_x, const … in compute_batch_to_space_shape() function