| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qembeddingbag_prepack.cpp | 52 uint8_t* weight_data = static_cast<uint8_t*>(weight_contig.data_ptr()); in prepack() local
258 const auto weight_data = in qembeddingbag_byte_prepack_out() local
270 const auto weight_data = weight_contig->data_ptr<float>(); in qembeddingbag_byte_prepack_out() local
286 const auto weight_data = float_weight.data_ptr<float>(); in qembeddingbag_byte_prepack_out() local
404 const auto weight_data = in _qembeddingbag_nbit_prepack_helper() local
417 const auto weight_data = weight_contig.data_ptr<float>(); in _qembeddingbag_nbit_prepack_helper() local
435 const auto weight_data = float_weight.data_ptr<float>(); in _qembeddingbag_nbit_prepack_helper() local
|
| D | Normalization.cpp | 28 const float* weight_data, in compute_fused_params()
83 const float* weight_data = weight.template const_data_ptr<float>(); in q_batch_norm1d_impl() local
192 const float* weight_data = weight.template const_data_ptr<float>(); in q_batch_norm2d_impl() local
288 const float* weight_data = weight.template const_data_ptr<float>(); in q_batch_norm3d_impl() local
|
| D | qembeddingbag.cpp | 49 const auto weight_data = weight.data_ptr<uint8_t>(); in embedding_lookup_fallback_impl() local
298 const auto weight_data = weight.data_ptr<uint8_t>(); in embedding_bag_nbit_impl() local
468 const auto weight_data = weight.data_ptr<uint8_t>(); in embedding_bag_byte_impl() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | LossMultiMargin.cpp | 24 const scalar_t* weight_data, in multi_margin_inner_sum_cpu()
66 const scalar_t* weight_data, in multi_margin_loss_cpu_kernel()
136 auto weight_data = in multi_margin_loss_out_cpu_template() local
160 const scalar_t* weight_data, in multi_margin_loss_backward_cpu_kernel()
241 auto weight_data = weight_contiguous.defined() in multi_margin_loss_backward_out_cpu_template() local
|
| D | LossNLL.cpp | 173 const scalar_t* weight_data = optional_data<const scalar_t>(weight_contiguous); in nll_loss_out_frame() local
356 const scalar_t* weight_data = optional_data<const scalar_t>(weight_contiguous); in nll_loss_backward_out_frame() local
|
| D | LossNLL2d.cpp | 116 const scalar_t* weight_data = optional_data<const scalar_t>(weight_contiguous); in nll_loss2d_forward_out_frame() local
293 const scalar_t* weight_data = optional_data<const scalar_t>(weight_contiguous); in nll_loss2d_backward_out_frame() local
|
| /external/executorch/backends/cadence/hifi/operators/ |
| D | quantized_linear_out.cpp | 49 const uint8_t* __restrict__ weight_data = weight.const_data_ptr<uint8_t>(); in _quantized_linear_asym8u() local
94 const int8_t* __restrict__ weight_data = weight.const_data_ptr<int8_t>(); in _quantized_linear_asym8s() local
139 const uint8_t* __restrict__ weight_data = weight.const_data_ptr<uint8_t>(); in _quantized_linear_per_tensor_asym8u() local
187 const int8_t* __restrict__ weight_data = weight.const_data_ptr<int8_t>(); in _quantized_linear_per_tensor_asym8s() local
|
| D | quantized_layer_norm.cpp | 43 const float* __restrict__ weight_data = weight.const_data_ptr<float>(); in quantized_layer_norm_per_tensor_() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected_test.cc | 1646 std::initializer_list<float> weight_data = { in TEST_P() local
1673 std::initializer_list<float> weight_data = { in TEST_P() local
1701 std::initializer_list<float> weight_data = { in TEST_P() local
1726 std::initializer_list<float> weight_data = { in TEST_P() local
1756 std::initializer_list<float> weight_data = { in TEST_P() local
1786 std::initializer_list<float> weight_data = { in TEST_P() local
1820 std::initializer_list<float> weight_data = { in TEST_P() local
1864 std::initializer_list<float> weight_data = { in TEST_P() local
1926 std::initializer_list<float> weight_data = { in TEST_P() local
2046 std::vector<float> weight_data = { in TEST_P() local
[all …]
|
| /external/executorch/backends/cadence/reference/operators/ |
| D | quantized_linear_out.cpp | 32 const T* __restrict__ weight_data = weight.const_data_ptr<T>(); in _typed_quantized_linear() local
|
| D | quantized_layer_norm.cpp | 40 const float* __restrict__ weight_data = weight.const_data_ptr<float>(); in quantized_layer_norm_per_tensor_() local
|
| /external/executorch/backends/arm/runtime/ |
| D | VelaBinStream.h | 55 const char* weight_data; member
|
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| D | NativeLayerNorm.cpp | 60 const ValueRef weight_data, in add_native_layer_norm_node()
|
| D | Linear.cpp | 276 ValueRef weight_data = args.at(1); in linear() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_dimension_numbers_test.cc | 92 auto weight_data = in XLA_TEST_F() local
|
| /external/executorch/kernels/test/ |
| D | op_convolution_backward_test.cpp | 99 std::vector<float> weight_data = { in TEST_F() local
|
| D | op_native_layer_norm_test.cpp | 64 const std::vector<ctype> weight_data; member
|
| /external/trusty/arm-trusted-firmware/include/plat/arm/common/ |
| D | fconf_ethosn_getter.h | 36 struct ethosn_allocator_t weight_data; member
|
| /external/executorch/kernels/portable/cpu/ |
| D | op_native_layer_norm.cpp | 56 const CTYPE* weight_data; in layer_norm() local
|
| D | op_native_group_norm.cpp | 62 const CTYPE* weight_data; in group_norm() local
|
| D | op_convolution_backward.cpp | 128 const CTYPE* weight_data = weight.const_data_ptr<CTYPE>(); in conv2d_backward_impl() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 218 T* weight_data = const_cast<T*>(weight_tensor.flat<T>().data()); in Compute() local
|
| D | mkl_matmul_ops_common.h | 126 void Execute(const Tinput* src_data, const Tweight* weight_data, in Execute()
503 Tweight* weight_data, const Tensor& weight_tensor, in CacheWeight()
|
| D | mkl_qmatmul_op.cc | 261 Tweight* weight_data = nullptr; in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bincount_op.cc | 108 const T* weight_data = weights.data(); in Compute() local
|