Searched refs:compute_winograd_output_transform_shape (Results 1 – 3 of 3) sorted by relevance
222 …TensorShape output_transform_shape = compute_winograd_output_transform_shape(TensorInfo(batched_ge… in compute_reference()524 …TensorShape output_shape = compute_winograd_output_transform_shape(TensorInfo(input_shape, 1, data… in compute_target()569 …TensorShape output_shape = compute_winograd_output_transform_shape(TensorInfo(input_shape,… in compute_reference()
91 …fo tensor_info_output = input->clone()->set_tensor_shape(compute_winograd_output_transform_shape(*… in validate_arguments()136 …auto_init_if_empty(*dst, src->clone()->set_tensor_shape(compute_winograd_output_transform_shape(*s… in configure()
677 inline TensorShape compute_winograd_output_transform_shape(const ITensorInfo &input, const Winograd… in compute_winograd_output_transform_shape() function