Home
last modified time | relevance | path

Searched refs:weight_array (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_dimension_numbers_test.cc93 auto weight_array = absl::make_unique<Array4D<float>>(4, 3, 1, 1); in XLA_TEST_F() local
94 weight_array->FillWithMultiples(0.2); in XLA_TEST_F()
96 client_->TransferToServer(LiteralUtil::CreateR4FromArray4D(*weight_array)) in XLA_TEST_F()
122 auto expected_conv1 = ReferenceUtil::ConvArray4D(*input_array, *weight_array, in XLA_TEST_F()
/external/libaom/libaom/aom_dsp/x86/
Dintrapred_ssse3.c610 static INLINE void load_weight_w4(const uint8_t *weight_array, int height, in load_weight_w4() argument
614 const __m128i t = _mm_cvtsi32_si128(((const uint32_t *)weight_array)[1]); in load_weight_w4()
620 const __m128i weight = _mm_loadl_epi64((const __m128i *)&weight_array[8]); in load_weight_w4()
624 const __m128i weight = _mm_loadu_si128((const __m128i *)&weight_array[16]); in load_weight_w4()
745 static INLINE void load_weight_w8(const uint8_t *weight_array, int height, in load_weight_w8() argument
749 __m128i we = _mm_loadu_si128((const __m128i *)&weight_array[we_offset]); in load_weight_w8()
766 we = _mm_loadu_si128((const __m128i *)&weight_array[16]); in load_weight_w8()
773 _mm_loadu_si128((const __m128i *)&weight_array[32]); in load_weight_w8()
779 _mm_loadu_si128((const __m128i *)&weight_array[32 + 16]); in load_weight_w8()
1030 static INLINE void load_weight_v_w4(const uint8_t *weight_array, int height, in load_weight_v_w4() argument
[all …]
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc97 Array& weight_array = input_model_.GetOrCreateArray("weights"); in BuildQuantizableTestModel() local
108 weight_array.data_type = ArrayDataType::kFloat; in BuildQuantizableTestModel()
111 Shape* weight_array_shape = weight_array.mutable_shape(); in BuildQuantizableTestModel()
116 weight_array.GetMutableBuffer<ArrayDataType::kFloat>(); in BuildQuantizableTestModel()
119 weight_array.GetMutableBuffer<ArrayDataType::kFloat>().data.data(); in BuildQuantizableTestModel()