Searched refs:compute_tiled_shape (Results 1 – 4 of 4) sorted by relevance
54 …UTE_RETURN_ERROR_ON_MISMATCHING_DIMENSIONS(misc::shape_calculator::compute_tiled_shape(input->tens… in validate_arguments()72 …TensorShape tiled_shape = misc::shape_calculator::compute_tiled_shape(input->info()->tensor_shape(… in configure()
49 …UTE_RETURN_ERROR_ON_MISMATCHING_DIMENSIONS(misc::shape_calculator::compute_tiled_shape(input->tens… in validate_arguments()73 …TensorShape tiled_shape = misc::shape_calculator::compute_tiled_shape(input->info()->tensor_shape(… in configure()
42 const auto tiled_shape = misc::shape_calculator::compute_tiled_shape(src.shape(), multiples); in tile()
1254 inline TensorShape compute_tiled_shape(const TensorShape &input_shape, const Multiples &multiples) in compute_tiled_shape() function