Home
last modified time | relevance | path

Searched refs:data_ptr_from_mkldnn (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/inductor/aoti_torch/
Dmkldnn_tensor.cpp13 void* data_ptr_from_mkldnn(at::Tensor* mkldnn_tensor) { in data_ptr_from_mkldnn() function
15 at::native::data_ptr_from_mkldnn(*mkldnn_tensor)); in data_ptr_from_mkldnn()
31 void* data_ptr_from_mkldnn(at::Tensor* mkldnn_tensor) {
Dmkldnn_tensor.h7 void* data_ptr_from_mkldnn(at::Tensor* mkldnn_tensor);
Dshim_common.cpp230 *ret_data_ptr = data_ptr_from_mkldnn(t); in aoti_torch_get_data_ptr()
/external/pytorch/aten/src/ATen/native/mkldnn/
DMKLDNNCommon.cpp65 int64_t data_ptr_from_mkldnn(const Tensor& mkldnn_tensor) { in data_ptr_from_mkldnn() function
206 TORCH_FN(data_ptr_from_mkldnn)); in TORCH_LIBRARY_IMPL()
DMKLDNNCommon.h31 TORCH_API int64_t data_ptr_from_mkldnn(const Tensor& mkldnn_tensor);