Home
last modified time | relevance | path

Searched refs:PackToUint32 (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dquantize_test.cc68 auto pack_input_vec = PackToUint32<uint8>(input_row); in PackLargeInput()
165 auto output = PackToUint32<uint8>(input); in TEST()
172 auto output = PackToUint32<int8>(input); in TEST()
178 auto output = PackToUint32<uint8>(input); in TEST()
185 auto x = ConstantR1<uint32>(&builder, PackToUint32<uint16>(input)); in XLA_TEST_F()
195 auto x = ConstantR1<uint32>(&builder, PackToUint32<uint8>(input)); in XLA_TEST_F()
210 auto x = ConstantR2<uint32>(&builder, {{PackToUint32<uint8>(input[0])[0]}, in XLA_TEST_F()
211 {PackToUint32<uint8>(input[1])[0]}, in XLA_TEST_F()
212 {PackToUint32<uint8>(input[2])[0]}, in XLA_TEST_F()
213 {PackToUint32<uint8>(input[3])[0]}}); in XLA_TEST_F()
[all …]
Dquantize.h50 inline std::vector<uint32> PackToUint32(absl::Span<const T> input) { in PackToUint32() function