Searched refs:input_transform_shape (Results 1 – 1 of 1) sorted by relevance
218 …TensorShape input_transform_shape = compute_winograd_input_transform_shape(TensorInfo(input_shape… in compute_reference() local220 TensorShape batched_gemm_shape = input_transform_shape; in compute_reference()228 …t_transform_out = reference::winograd_input_transform<T1>(src_t1, input_transform_shape, winograd_… in compute_reference()