Home
last modified time | relevance | path

Searched defs:weight_data (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqembeddingbag_prepack.cpp52 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
H A DNormalization.cpp28 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
H A Dqembeddingbag.cpp49 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dfully_connected_test.cc1646 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 …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLossMultiMargin.cpp24 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
H A DLossNLL2d.cpp116 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
H A DLossNLL.cpp173 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
/aosp_15_r20/external/executorch/backends/cadence/hifi/operators/
H A Dquantized_linear_out.cpp49 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
H A Dquantized_layer_norm.cpp43 const float* __restrict__ weight_data = weight.const_data_ptr<float>(); in quantized_layer_norm_per_tensor_() local
/aosp_15_r20/external/executorch/backends/cadence/reference/operators/
H A Dquantized_linear_out.cpp32 const T* __restrict__ weight_data = weight.const_data_ptr<T>(); in _typed_quantized_linear() local
H A Dquantized_layer_norm.cpp40 const float* __restrict__ weight_data = weight.const_data_ptr<float>(); in quantized_layer_norm_per_tensor_() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DNativeLayerNorm.cpp60 const ValueRef weight_data, in add_native_layer_norm_node()
H A DLinear.cpp276 ValueRef weight_data = args.at(1); in linear() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconvolution_dimension_numbers_test.cc92 auto weight_data = in XLA_TEST_F() local
/aosp_15_r20/external/executorch/backends/arm/runtime/
H A DVelaBinStream.h55 const char* weight_data; member
/aosp_15_r20/external/executorch/kernels/test/
H A Dop_convolution_backward_test.cpp99 std::vector<float> weight_data = { in TEST_F() local
H A Dop_native_layer_norm_test.cpp64 const std::vector<ctype> weight_data; member
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_matmul_ops_common.h126 void Execute(const Tinput* src_data, const Tweight* weight_data, in Execute()
503 Tweight* weight_data, const Tensor& weight_tensor, in CacheWeight()
H A Dmkl_matmul_op_fused.cc218 T* weight_data = const_cast<T*>(weight_tensor.flat<T>().data()); in Compute() local
H A Dmkl_qmatmul_op.cc261 Tweight* weight_data = nullptr; in Compute() local
/aosp_15_r20/external/trusty/arm-trusted-firmware/include/plat/arm/common/
Dfconf_ethosn_getter.h36 struct ethosn_allocator_t weight_data; member
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_native_layer_norm.cpp56 const CTYPE* weight_data; in layer_norm() local
H A Dop_native_group_norm.cpp62 const CTYPE* weight_data; in group_norm() local
H A Dop_convolution_backward.cpp128 const CTYPE* weight_data = weight.const_data_ptr<CTYPE>(); in conv2d_backward_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dbatch_norm_kernel.cpp37 const param_t* weight_data = weight.defined() ? weight.const_data_ptr<param_t>() : nullptr; in batch_norm_cpu_collect_linear_and_constant_terms() local
553 const scalar_t* weight_data = weight_.const_data_ptr<scalar_t>(); in batch_norm_cpu_backward_channels_last_impl() local
1149 opmath_t* weight_data = weight_f.data_ptr<opmath_t>(); in batch_norm_cpu_backward_channels_last_internal() local

12