Searched refs:weight_array (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | convolution_dimension_numbers_test.cc | 90 auto weight_array = std::make_unique<Array4D<float>>(4, 3, 1, 1); in XLA_TEST_F() local
91 weight_array->FillWithMultiples(0.2); in XLA_TEST_F()
93 client_->TransferToServer(LiteralUtil::CreateR4FromArray4D(*weight_array)) in XLA_TEST_F()
119 auto expected_conv1 = ReferenceUtil::ConvArray4D(*input_array, *weight_array, in XLA_TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/ |
| H A D | export_test.cc | 109 Array& weight_array = input_model_.GetOrCreateArray("weights"); in BuildQuantizableTestModel() local
120 weight_array.data_type = ArrayDataType::kFloat; in BuildQuantizableTestModel()
123 Shape* weight_array_shape = weight_array.mutable_shape(); in BuildQuantizableTestModel()
128 weight_array.GetMutableBuffer<ArrayDataType::kFloat>(); in BuildQuantizableTestModel()
131 weight_array.GetMutableBuffer<ArrayDataType::kFloat>().data.data(); in BuildQuantizableTestModel()
|
| /aosp_15_r20/external/libgav1/src/dsp/x86/ |
| H A D | intrapred_smooth_sse4.cc | 149 inline void LoadSmoothWeights4(const uint8_t* LIBGAV1_RESTRICT weight_array, in LoadSmoothWeights4() argument
153 const __m128i x_weights = Load4(weight_array); in LoadSmoothWeights4()
159 const __m128i y_weights = LoadLo8(weight_array + 4); in LoadSmoothWeights4()
164 const __m128i y_weights = LoadUnaligned16(weight_array + 12); in LoadSmoothWeights4()
317 inline void LoadSmoothWeights8(const uint8_t* LIBGAV1_RESTRICT weight_array, in LoadSmoothWeights8() argument
321 __m128i loaded_weights = LoadUnaligned16(&weight_array[offset]); in LoadSmoothWeights8()
339 loaded_weights = LoadUnaligned16(weight_array + 12); in LoadSmoothWeights8()
346 const __m128i weight_lo = LoadUnaligned16(weight_array + 28); in LoadSmoothWeights8()
351 const __m128i weight_hi = LoadUnaligned16(weight_array + 44); in LoadSmoothWeights8()
1421 weight_array, in LoadSmoothVerticalWeights4() argument
[all …]
|
| /aosp_15_r20/external/libaom/aom_dsp/x86/ |
| H A D | intrapred_ssse3.c | 1147 const uint8_t *LIBAOM_RESTRICT weight_array, const int height, in load_smooth_vertical_weights4() argument
1152 const __m128i weight = Load4(weight_array); in load_smooth_vertical_weights4()
1156 const __m128i weight = LoadLo8(weight_array + 4); in load_smooth_vertical_weights4()
1160 const __m128i weight = LoadUnaligned16(weight_array + 12); in load_smooth_vertical_weights4()
|