| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Conv.cpp | 167 const Tensor& weight_t, in _mkldnn_convolution_out()
215 const Tensor& weight_t, in _mkldnn_convolution()
280 const Tensor& weight_t, in mkldnn_convolution()
301 const Tensor& weight_t, in mkldnn_convolution_pointwise()
338 const Tensor& weight_t, in mkldnn_convolution_pointwise_binary()
502 const Tensor& weight_t, in mkldnn_convolution_pointwise_binary_()
593 const Tensor& weight_t, in _original_deconv_weight_size()
620 const Tensor& weight_t, in _mkldnn_convolution_transpose()
716 const Tensor& weight_t, in mkldnn_convolution_transpose_pointwise()
747 const Tensor& weight_t, in mkldnn_convolution_transpose_pointwise_meta()
[all …]
|
| D | Linear.cpp | 61 const Tensor& weight_t, const std::optional<Tensor>& bias_opt) { in mkldnn_linear()
112 IntArrayRef input_size, const Tensor& grad_output, const Tensor& weight_t){ in mkldnn_linear_backward_input()
185 const Tensor& weight_t, in mkldnn_linear_pointwise()
260 const Tensor& weight_t, in mkldnn_linear_pointwise_binary()
|
| D | RNN.cpp | 199 auto weight_t = weight.contiguous(); in _shuffle_weight() local
|
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | ConvShared.cpp | 242 const Tensor& weight_t, in cudnn_convolution()
278 const Tensor& weight_t, in cudnn_convolution_out()
311 const Tensor& weight_t, in cudnn_convolution_transpose_backward_input()
444 const Tensor& weight_t, in cudnn_convolution_backward_input()
470 const Tensor& weight_t, in cudnn_convolution_transpose()
707 const Tensor& weight_t, in cudnn_convolution_relu()
759 const Tensor& weight_t, in cudnn_convolution_add_relu()
|
| D | ConvPlaceholders.cpp | 43 const Tensor& weight_t, in cudnn_convolution_out()
207 const Tensor& weight_t, in cudnn_convolution_relu()
218 const Tensor& weight_t, in cudnn_convolution_add_relu()
|
| D | BatchNorm.cpp | 123 const Tensor& weight_t, in cudnn_batch_norm()
274 const Tensor& weight_t, in cudnn_batch_norm_backward()
|
| /external/sdv/vsomeip/implementation/service_discovery/include/ |
| D | primitive_types.hpp | 16 typedef uint16_t weight_t; typedef
|
| /external/pytorch/aten/src/ATen/native/miopen/ |
| D | BatchNorm_miopen.cpp | 61 …const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, const std:… in miopen_batch_norm()
162 const Tensor& weight_t, in miopen_batch_norm_backward()
|
| D | Conv_miopen.cpp | 798 const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, in miopen_convolution()
899 const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, in miopen_depthwise_convolution()
1168 const Tensor& grad_output_t, const Tensor& weight_t, in miopen_convolution_transpose_backward_input()
1317 IntArrayRef input_size, const Tensor& grad_output_t, const Tensor& weight_t, in miopen_convolution_backward_input()
1411 IntArrayRef input_size, const Tensor& grad_output_t, const Tensor& weight_t, in miopen_depthwise_convolution_backward_input()
1466 const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, in miopen_convolution_transpose()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | frozen_conv_add_relu_fusion_cuda.cpp | 85 const at::Tensor& weight_t = weight.value().toTensor(); in fuseFrozenConvAddReluImpl() local
|
| D | frozen_linear_transpose.cpp | 63 Value* weight_t = graph_->insertConstant(std::move(weight_t_tensor)); in replace_linear_with_matmul() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 507 const Tensor& weight_t = weight_oi_; in CacheWeight() local
547 const Tensor& weight_t = weight_oi_; in GetCachedWeight() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | mkl_remapper_test.cc | 783 Tensor weight_t = Tensor(DataTypeToEnum<T>::v(), weight_shape); in VerifyFused() local
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 1293 const Tensor& weight_t, in convolution_jvp()
1331 const Tensor& weight_t, in _convolution_jvp()
6158 const Tensor& weight_t, in _affine_jvp()
6193 const Tensor& weight_t, in batch_norm_jvp()
6248 const Tensor& weight_t, in layer_norm_jvp()
6287 const Tensor& weight_t, in group_norm_jvp()
|