Searched defs:mkldnn_tensor (Results 1 – 4 of 4) sorted by relevance
65 int64_t data_ptr_from_mkldnn(const Tensor& mkldnn_tensor) { in data_ptr_from_mkldnn()105 ideep::tensor& itensor_from_mkldnn(const MKLDNNTensor& mkldnn_tensor) { in itensor_from_mkldnn()112 int64_t nbytes_from_mkldnn(const Tensor& mkldnn_tensor) { in nbytes_from_mkldnn()
29 Tensor mkldnn_to_dense(const Tensor& mkldnn_tensor, std::optional<ScalarType> dtype, std::optional<… in mkldnn_to_dense()
609 Node* createConstantMKLDNNTensorOp(Graph* g, const Tensor& mkldnn_tensor) { in createConstantMKLDNNTensorOp()623 auto mkldnn_tensor = constant_as<Tensor>(input)->to_mkldnn(); in replaceInputWithMKLDNNTensor() local635 const at::Tensor& mkldnn_tensor) { in replaceInputWithMKLDNNTensor()647 auto mkldnn_tensor = constant_as<Tensor>(input)->to_mkldnn(); in replaceInputWithMKLDNNTensor() local
13 void* data_ptr_from_mkldnn(at::Tensor* mkldnn_tensor) { in data_ptr_from_mkldnn()