Searched refs:cpu_tensor_cont (Results 1 – 1 of 1) sorted by relevance
92 auto cpu_tensor_cont = cpu_tensor.contiguous(); in dense_to_mkldnn() local105 Tensor mkldnn_tensor = empty_mkldnn(cpu_tensor_cont.sizes(), data_type, in dense_to_mkldnn()106 … cpu_tensor_cont.options().layout_opt(), cpu_tensor_cont.options().device_opt(), in dense_to_mkldnn()107 cpu_tensor_cont.options().pinned_memory_opt()); in dense_to_mkldnn()112 (cpu_tensor_cont.template data_ptr<float>())); in dense_to_mkldnn()116 cpu_tensor_cont.template data_ptr<BFloat16>()); in dense_to_mkldnn()120 cpu_tensor_cont.template data_ptr<Half>()); in dense_to_mkldnn()124 cpu_tensor_cont.template data_ptr<uint8_t>()); in dense_to_mkldnn()130 cpu_tensor_cont.template data_ptr<int8_t>()); in dense_to_mkldnn()