Searched defs:weight_buf (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_external_calls.cpp | 68 std::vector<float> weight_buf(100 * 1 * 7, 6.f); in TEST() local
122 std::vector<int32_t> weight_buf(100 * 1 * 7, 6); in TEST() local
164 std::vector<float> weight_buf(10 * 1 * 7, 6.f); in TEST() local
222 std::vector<float> weight_buf(16 * 3 * 3 * 3, 6.f); in TEST() local
283 std::vector<int32_t> weight_buf(16 * 3 * 3 * 3, 6); in TEST() local
325 std::vector<float> weight_buf(16 * 16 * 1 * 1, 6.f); in TEST() local
420 std::vector<float> weight_buf(256 * 100, 5.f); in TEST() local
844 std::vector<float> weight_buf(16 * 16 * 1 * 1, 6.f); in TEST() local
|
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | RNN.cpp | 74 const Tensor& weight_buf, in _cudnn_rnn_backward()
597 const Tensor& weight_buf, in add_projection_weights()
705 const Tensor& weight_buf, in get_parameters()
875 const Tensor& weight_buf, in get_parameters()
1296 auto weight_buf = at::zeros(num_weights, flat_buf_options); in copy_weights_to_flat_buf_views() local
1408 auto weight_buf = weight_buf_r; in _cudnn_rnn() local
1671 const Tensor& weight_buf, in _cudnn_rnn_backward_input()
1904 const Tensor& weight_buf, in _cudnn_rnn_backward_weight()
2092 const Tensor& weight_buf, in _cudnn_rnn_backward()
2446 auto weight_buf = at::empty({0}, any_param.options()).set_(param_storage); in try_get_weight_buf() local
[all …]
|
| /external/pytorch/aten/src/ATen/cudnn/ |
| D | AutocastRNN.cpp | 60 Tensor weight_buf, redispatch_weight_buf; in _cudnn_rnn_cast_reflatten() local
|
| /external/pytorch/aten/src/ATen/native/miopen/ |
| D | RNN_miopen.cpp | 41 …const Tensor& input, TensorList weight, int64_t weight_stride0, const Tensor& weight_buf, const Te… in miopen_rnn_backward()
332 const Tensor& weight_buf) in get_parameters()
500 auto weight_buf = at::empty(num_weights, x.options()); in miopen_rnn() local
558 const Tensor& input_r, const Tensor& weight_buf, const Tensor& hx, const Tensor& cx, in miopen_rnn_backward_input()
674 const Tensor& weight_buf, const Tensor& hx, const Tensor& cx, in miopen_rnn_backward_weight()
761 …const Tensor& input, TensorList weight, int64_t weight_stride0, const Tensor& weight_buf, const Te… in miopen_rnn_backward()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | export_test.cc | 114 auto weight_buf = std::make_unique<float[]>(buf_size); in BuildQuantizableTestModel() local
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 5921 weight_buf, argument
|