Searched refs:compute_winograd_input_transform_shape (Results 1 – 4 of 4) sorted by relevance
68 …const TensorShape output_shape = misc::shape_calculator::compute_winograd_input_transform_shape(*i… in validate_arguments()138 …const TensorShape output_shape = misc::shape_calculator::compute_winograd_input_transform_shape(*s… in configure()
218 …TensorShape input_transform_shape = compute_winograd_input_transform_shape(TensorInfo(input_shape… in compute_reference()249 …TensorShape output_shape = compute_winograd_input_transform_shape(TensorInfo(input_shape, 1, data_… in setup()
135 …const TensorShape input0_shape = misc::shape_calculator::compute_winograd_input_transform_shape(*s… in validate_arguments()
641 inline TensorShape compute_winograd_input_transform_shape(const ITensorInfo &input, const WinogradI… in compute_winograd_input_transform_shape() function