Searched defs:output_reshaped (Results 1 – 9 of 9) sorted by relevance
136 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local176 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
164 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
135 auto output_reshaped = output->flat<T>(); in Compute() local254 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute() local
103 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute() local
162 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute() local
200 auto output_reshaped = out->template flat_inner_dims<Scalar, 1>(); in ComputeAsync() local
579 Tensor output_reshaped; in ContractOperands() local
172 auto output_reshaped = NCHWToNCHW_VECT_C(output); in Compile() local
176 auto output_reshaped = NCHWToNCHW_VECT_C(output); in Compile() local