Searched defs:weight_arg (Results 1 – 10 of 10) sorted by relevance
| /external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| D | Lerp.cpp | 49 const Scalar& weight_arg) { in _lerp_scalar()
112 const Scalar& weight_arg) { in _lerp_scalar_()
168 const Tensor& weight_arg) { in _lerp_tensor()
241 const Tensor& weight_arg) { in _lerp_tensor_()
|
| D | Mm.cpp | 50 vTensor pack_weights_using_height_packing(const Tensor& weight_arg) { in pack_weights_using_height_packing()
84 vTensor pack_weights(const Tensor& weight_arg, const bool use_batch = false) { in pack_weights()
151 const Tensor& weight_arg, in pack_biases()
168 const Tensor& weight_arg, in pack_biases_quantized_weights()
|
| D | Convolution.cpp | 520 const Tensor weight_arg = quantized ? at::dequantize(weight_inp) : weight_inp; in pack_weights() local
794 static vTensor pack_weights_using_width_packing(const Tensor& weight_arg) { in pack_weights_using_width_packing()
822 const Tensor& weight_arg, in run_conv1d_context_impl()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | NaiveConvolutionTranspose2d.cu | 159 weight_arg{weight, "weight", 3}, bias_arg{bias, "bias", 4}; in slow_conv_transpose2d_out_cuda_template() local
340 weight_arg{weight_, "weight", 3}, in slow_conv_transpose2d_backward_out_cuda_template() local
|
| D | NaiveConvolutionTranspose3d.cu | 228 weight_arg{weight_, "weight", 3}, bias_arg{bias_, "bias", 4}; in slow_conv_transpose3d_out_cuda_template() local
450 weight_arg{weight_, "weight", 3}, in slow_conv_transpose3d_backward_out_cuda_template() local
|
| D | NaiveDilatedConvolution.cu | 157 TensorArg input_arg{input, "input", 2}, weight_arg{weight, "weight", 3}, in slow_conv_dilated_location_check() local
|
| D | EmbeddingBag.cu | 359 auto weight_arg = TensorArg(weight, "weight", 1); in _embedding_bag_cuda() local
|
| /external/grpc-grpc/src/core/load_balancing/ring_hash/ |
| D | ring_hash.cc | 386 auto weight_arg = endpoint.args().GetInt(GRPC_ARG_ADDRESS_WEIGHT); in Ring() local
637 int weight_arg = in UpdateLocked() local
|
| /external/pytorch/aten/src/ATen/native/miopen/ |
| D | Conv_miopen.cpp | 1571 weight_arg { weight, "weight", 2 }; in miopen_convolution_add_relu() local
1663 weight_arg { weight, "weight", 2 }; in miopen_convolution_relu() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | EmbeddingBag.cpp | 882 auto weight_arg = TensorArg(weight, "weight", 1); in check_arguments() local
|