Searched refs:compute_space_to_batch_shape (Results 1 – 3 of 3) sorted by relevance
78 …TensorShape expected_output_shape = misc::shape_calculator::compute_space_to_batch_shape(input, bl… in validate_arguments_static()114 …TensorShape output_shape = misc::shape_calculator::compute_space_to_batch_shape(input->info(), blo… in configure()
74 …TensorShape expected_output_shape = misc::shape_calculator::compute_space_to_batch_shape(input, bl… in validate_arguments_static()140 …TensorShape output_shape = misc::shape_calculator::compute_space_to_batch_shape(input->info(), blo… in configure()
1165 inline TensorShape compute_space_to_batch_shape(const ITensorInfo *input, const int block_x, const … in compute_space_to_batch_shape() function