Searched refs:input_cl (Results 1 – 2 of 2) sorted by relevance
310 at::Tensor input_cl = input.contiguous(at::MemoryFormat::ChannelsLast); in upsample_bilinear2d_out_cuda_template() local312 const scalar_t* idata = input_cl.const_data_ptr<scalar_t>(); in upsample_bilinear2d_out_cuda_template()
8218 input_cl = input.to(memory_format=memory_format)8219 out_cl2 = conv(input_cl)8221 out_cl3 = conv_cl(input_cl)