Home
last modified time | relevance | path

Searched refs:itensor_from_tensor (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DConv.cpp180 const ideep::tensor x = itensor_from_tensor(input, /*from_const_data_ptr*/true); in _mkldnn_convolution_out()
181 const ideep::tensor w = itensor_from_tensor(weight, /*from_const_data_ptr*/true); in _mkldnn_convolution_out()
183 const ideep::tensor b = itensor_from_tensor(bias, /*from_const_data_ptr*/true); in _mkldnn_convolution_out()
254 y = itensor_from_tensor(output); in _mkldnn_convolution()
408 const ideep::tensor x = itensor_from_tensor(input); in mkldnn_convolution_pointwise_binary()
409 const ideep::tensor w = itensor_from_tensor(weight); in mkldnn_convolution_pointwise_binary()
410 const ideep::tensor z = itensor_from_tensor(other); in mkldnn_convolution_pointwise_binary()
411 ideep::tensor y = itensor_from_tensor(output); in mkldnn_convolution_pointwise_binary()
429 const ideep::tensor b = itensor_from_tensor(bias); in mkldnn_convolution_pointwise_binary()
553 ideep::tensor y = itensor_from_tensor(other_t); in mkldnn_convolution_pointwise_binary_()
[all …]
H A DLinear.cpp88 const ideep::tensor w = itensor_from_tensor(weight); in mkldnn_linear()
92 const ideep::tensor b = itensor_from_tensor(bias); in mkldnn_linear()
213 ideep::tensor mkldnn_output = itensor_from_tensor(output); in mkldnn_linear_pointwise()
223 mkldnn_bias = itensor_from_tensor(bias); in mkldnn_linear_pointwise()
225 const ideep::tensor w = itensor_from_tensor(weight_t); in mkldnn_linear_pointwise()
305 ideep::tensor mkldnn_output = itensor_from_tensor(output); in mkldnn_linear_pointwise_binary()
306 const ideep::tensor mkldnn_other = itensor_from_tensor(other_reshaped); in mkldnn_linear_pointwise_binary()
311 mkldnn_bias = itensor_from_tensor(bias); in mkldnn_linear_pointwise_binary()
313 const ideep::tensor w = itensor_from_tensor(weight_t); in mkldnn_linear_pointwise_binary()
H A DNormalization.cpp153 ideep::tensor w = itensor_from_tensor(weight); in mkldnn_batch_norm()
154 ideep::tensor b = itensor_from_tensor(bias); in mkldnn_batch_norm()
170 ideep::tensor m = itensor_from_tensor(running_mean); in mkldnn_batch_norm()
171 ideep::tensor v = itensor_from_tensor(running_var); in mkldnn_batch_norm()
191 ideep::tensor m = itensor_from_tensor(running_mean); in mkldnn_batch_norm()
192 ideep::tensor v = itensor_from_tensor(running_var); in mkldnn_batch_norm()
262 ideep::tensor w = itensor_from_tensor(weight); in mkldnn_batch_norm_backward()
H A DGelu.cpp41 const ideep::tensor& x = itensor_from_tensor(input); in mkldnn_gelu()
52 const ideep::tensor& x = itensor_from_tensor(input); in mkldnn_gelu_backward()
53 ideep::tensor grady = itensor_from_tensor(grad_output); in mkldnn_gelu_backward()
H A DMKLDNNConversions.cpp164 auto w = itensor_from_tensor(self_); in mkldnn_reorder_conv2d_weight()
221 auto w = itensor_from_tensor(self_); in mkldnn_reorder_conv3d_weight()
266 auto w = itensor_from_tensor(self_); in mkldnn_reorder_linear_weight()
357 ideep::tensor w = itensor_from_tensor(self_); in mkldnn_reorder_conv_transpose_weight()
523 const ideep::tensor packed_w = itensor_from_tensor(self); in get_mkldnn_serialized_md()
H A DPrelu.cpp34 const ideep::tensor& w = itensor_from_tensor(weight); in mkldnn_prelu()
45 const ideep::tensor& w = itensor_from_tensor(weight); in mkldnn_prelu_backward()
H A DConvPrepack.cpp162 const ideep::tensor mkldnn_input = itensor_from_tensor(input); in mkldnn_convolution_out()
165 mkldnn_bias = itensor_from_tensor(bias); in mkldnn_convolution_out()
206 ideep::tensor mkldnn_output = itensor_from_tensor(output); in run()
H A DMKLDNNCommon.h60 TORCH_API ideep::tensor itensor_from_tensor(const Tensor& tensor, bool from_const_data_ptr=false);
H A DRNN.cpp274 …w1_ = weight_ih.is_mkldnn() ? itensor_from_tensor(weight_ih) : itensor_view_from_dense(weight_ih, … in mkldnn_rnn_layer()
275 …w2_ = weight_hh.is_mkldnn() ? itensor_from_tensor(weight_hh) : itensor_view_from_dense(weight_hh, … in mkldnn_rnn_layer()
H A DMKLDNNCommon.cpp191 ideep::tensor itensor_from_tensor(const Tensor& tensor, bool from_const_data_ptr) { in itensor_from_tensor() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DActivation.cpp395 const ideep::tensor& x = itensor_from_tensor(self, /*from_const_data_ptr*/true); in TORCH_IMPL_FUNC()
396 ideep::tensor y = itensor_from_tensor(result); in TORCH_IMPL_FUNC()
401 const ideep::tensor& x = itensor_from_tensor(self); in TORCH_IMPL_FUNC()
402 ideep::tensor y = itensor_from_tensor(result); in TORCH_IMPL_FUNC()
420 const ideep::tensor& x = itensor_from_tensor(self, /*from_const_data_ptr*/true); in TORCH_IMPL_FUNC()
421 ideep::tensor grady = itensor_from_tensor(grad, /*from_const_data_ptr*/true); in TORCH_IMPL_FUNC()
422 ideep::tensor gradx = itensor_from_tensor(grad_input); in TORCH_IMPL_FUNC()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear.cpp985 auto src = at::native::itensor_from_tensor(input_contig); in linear_int8_with_onednn_weight()
1079 tensor wei_scales_t = at::native::itensor_from_tensor(weight_scales); in linear_int8_with_onednn_weight()