Home
last modified time | relevance | path

Searched refs:new_with_itensor_mkldnn (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DNormalization.cpp124 auto dst = at::native::new_with_itensor_mkldnn( in mkldnn_layer_norm_last_index_weight_bias_f32()
181 new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(input.options().dtype_opt()), in mkldnn_batch_norm()
183 …new_with_itensor_mkldnn(std::move(saved_mean), optTypeMetaToScalarType(weight.options().dtype_opt(… in mkldnn_batch_norm()
185 …new_with_itensor_mkldnn(std::move(saved_var), optTypeMetaToScalarType(weight.options().dtype_opt()… in mkldnn_batch_norm()
201 new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(input.options().dtype_opt()), in mkldnn_batch_norm()
203 … new_with_itensor_mkldnn(ideep::tensor{}, optTypeMetaToScalarType(weight.options().dtype_opt()), in mkldnn_batch_norm()
205 … new_with_itensor_mkldnn(ideep::tensor{}, optTypeMetaToScalarType(weight.options().dtype_opt()), in mkldnn_batch_norm()
272 … new_with_itensor_mkldnn(std::move(gradx), optTypeMetaToScalarType(input.options().dtype_opt()), in mkldnn_batch_norm_backward()
274 mkldnn_to_dense(new_with_itensor_mkldnn(std::move(gradw), in mkldnn_batch_norm_backward()
277 mkldnn_to_dense(new_with_itensor_mkldnn(std::move(gradb), in mkldnn_batch_norm_backward()
H A DPrelu.cpp39 return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(input.options().dtype_opt()), in mkldnn_prelu()
54 new_with_itensor_mkldnn(std::move(gradx), in mkldnn_prelu_backward()
57 new_with_itensor_mkldnn(std::move(gradw), in mkldnn_prelu_backward()
62 new_with_itensor_mkldnn(std::move(gradx), in mkldnn_prelu_backward()
65 mkldnn_to_dense(new_with_itensor_mkldnn(std::move(gradw), in mkldnn_prelu_backward()
H A DLinear.cpp103 … return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(self.options().dtype_opt()), in mkldnn_linear()
106 return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(self.options().dtype_opt()), in mkldnn_linear()
134 …return new_with_itensor_mkldnn(std::move(gradx), optTypeMetaToScalarType(grad_output.options().dty… in mkldnn_linear_backward_input()
137 …return new_with_itensor_mkldnn(std::move(gradx), optTypeMetaToScalarType(grad_output.options().dty… in mkldnn_linear_backward_input()
162 mkldnn_to_dense(new_with_itensor_mkldnn(std::move(gradw), in mkldnn_linear_backward_weights()
165 mkldnn_to_dense(new_with_itensor_mkldnn(std::move(gradb), in mkldnn_linear_backward_weights()
H A DTensorShape.cpp64 return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(self.options().dtype_opt()), in mkldnn_reshape()
76 …return new_with_itensor_mkldnn(std::move(dst), optTypeMetaToScalarType(self.options().dtype_opt()), in mkldnn_clone()
90 return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(self.options().dtype_opt()), in mkldnn_transpose()
H A DMKLDNNConversions.cpp194 …return new_with_itensor_mkldnn(std::move(result), optTypeMetaToScalarType(self.options().dtype_opt… in mkldnn_reorder_conv2d_weight()
241 …return new_with_itensor_mkldnn(std::move(result), optTypeMetaToScalarType(self.options().dtype_opt… in mkldnn_reorder_conv3d_weight()
280 …return new_with_itensor_mkldnn(std::move(result), optTypeMetaToScalarType(self.options().dtype_opt… in mkldnn_reorder_linear_weight()
383 …return new_with_itensor_mkldnn(std::move(result), optTypeMetaToScalarType(self.options().dtype_opt… in mkldnn_reorder_conv_transpose_weight()
511 …packed_w1 = new_with_itensor_mkldnn(std::move(w1_), optTypeMetaToScalarType(weight0.options().dtyp… in mkldnn_reorder_mkldnn_rnn_layer_weight()
517 …packed_w2 = new_with_itensor_mkldnn(std::move(w2_), optTypeMetaToScalarType(weight1.options().dtyp… in mkldnn_reorder_mkldnn_rnn_layer_weight()
H A DGelu.cpp45 return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(input.options().dtype_opt()), in mkldnn_gelu()
57 return new_with_itensor_mkldnn(std::move(gradx), in mkldnn_gelu_backward()
H A DRelu.cpp47 return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(input.options().dtype_opt()), in mkldnn_relu()
69 return new_with_itensor_mkldnn(std::move(gradx), in mkldnn_relu_backward()
H A DUnaryOps.cpp48 return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(self.options().dtype_opt()), in mkldnn_sigmoid()
64 return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(self.options().dtype_opt()), in mkldnn_tanh()
H A DMKLDNNCommon.cpp89 return at::native::new_with_itensor_mkldnn(std::move(a), dtype, device); in mkldnn_tensor_from_data_ptr()
92 Tensor new_with_itensor_mkldnn(ideep::tensor&& it, std::optional<ScalarType> dtype, std::optional<D… in new_with_itensor_mkldnn() function
H A DSoftMax.cpp44 return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(self.options().dtype_opt()), in mkldnn_softmax()
H A DTensorFactories.cpp24 return new_with_itensor_mkldnn(std::move(it), dtype, device); in empty_mkldnn()
H A DMKLDNNCommon.h42 TORCH_API Tensor new_with_itensor_mkldnn(ideep::tensor&& it, std::optional<ScalarType> dtype, std::…
H A DPooling.cpp287 …return new_with_itensor_mkldnn(std::move(y), optTypeMetaToScalarType(input.options().dtype_opt()),… in _mkldnn_pooling()
361 return new_with_itensor_mkldnn(std::move(gradx), in _mkldnn_pooling_backward()
H A DBinaryOps.cpp113 return new_with_itensor_mkldnn(std::move(z), optTypeMetaToScalarType(self.options().dtype_opt()), in mkldnn_add()
H A DConv.cpp121 new_with_itensor_mkldnn( \
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_ops_to_mkldnn.cpp264 out = at::native::new_with_itensor_mkldnn( in createUnaryOp()
378 auto a_out = at::native::new_with_itensor_mkldnn( in BroadOp()
384 auto b_out = at::native::new_with_itensor_mkldnn( in BroadOp()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear_prepack.cpp305 auto packed_weight = at::native::new_with_itensor_mkldnn( in pack_weight_to_onednn_tensor()
H A Dqconv_prepack.cpp608 auto packed_weight = at::native::new_with_itensor_mkldnn( in _qconv_prepack_onednn()