Home
last modified time | relevance | path

Searched refs:a_data (Results 1 – 25 of 50) sorted by relevance

12

/external/XNNPACK/test/
Dvbinaryc-microkernel-tester.h107 const uint16_t* a_data = inplace() ? y.data() : a.data(); in Test() local
113 y_ref[i] = fp16_ieee_to_fp32_value(a_data[i]) + fp16_ieee_to_fp32_value(b); in Test()
116 y_ref[i] = fp16_ieee_to_fp32_value(a_data[i]) / fp16_ieee_to_fp32_value(b); in Test()
119 y_ref[i] = fp16_ieee_to_fp32_value(b) / fp16_ieee_to_fp32_value(a_data[i]); in Test()
122 … y_ref[i] = std::max<float>(fp16_ieee_to_fp32_value(a_data[i]), fp16_ieee_to_fp32_value(b)); in Test()
125 … y_ref[i] = std::min<float>(fp16_ieee_to_fp32_value(a_data[i]), fp16_ieee_to_fp32_value(b)); in Test()
128 y_ref[i] = fp16_ieee_to_fp32_value(a_data[i]) * fp16_ieee_to_fp32_value(b); in Test()
132 const float diff = fp16_ieee_to_fp32_value(a_data[i]) - fp16_ieee_to_fp32_value(b); in Test()
137 y_ref[i] = fp16_ieee_to_fp32_value(a_data[i]) - fp16_ieee_to_fp32_value(b); in Test()
140 y_ref[i] = fp16_ieee_to_fp32_value(b) - fp16_ieee_to_fp32_value(a_data[i]); in Test()
[all …]
Dvbinary-microkernel-tester.h115 const uint16_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
122 y_ref[i] = fp16_ieee_to_fp32_value(a_data[i]) + fp16_ieee_to_fp32_value(b_data[i]); in Test()
125 y_ref[i] = fp16_ieee_to_fp32_value(a_data[i]) / fp16_ieee_to_fp32_value(b_data[i]); in Test()
128 …y_ref[i] = std::max<float>(fp16_ieee_to_fp32_value(a_data[i]), fp16_ieee_to_fp32_value(b_data[i])); in Test()
131 …y_ref[i] = std::min<float>(fp16_ieee_to_fp32_value(a_data[i]), fp16_ieee_to_fp32_value(b_data[i])); in Test()
134 y_ref[i] = fp16_ieee_to_fp32_value(a_data[i]) * fp16_ieee_to_fp32_value(b_data[i]); in Test()
138 … const float diff = fp16_ieee_to_fp32_value(a_data[i]) - fp16_ieee_to_fp32_value(b_data[i]); in Test()
143 y_ref[i] = fp16_ieee_to_fp32_value(a_data[i]) - fp16_ieee_to_fp32_value(b_data[i]); in Test()
149 vbinary(batch_size() * sizeof(uint16_t), a_data, b_data, y.data(), nullptr); in Test()
177 const uint16_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
[all …]
Dvadd-microkernel-tester.h168 const uint8_t* a_data = inplace_a() ? y.data() : a.data(); variable
196 float(int32_t(a_data[i]) - int32_t(a_zero_point())) * (a_scale() / y_scale()) +
200 y_ref[i] = xnn_qu8_quantize_add(a_data[i], b_data[i], scalar_quantization_params);
204 vadd_minmax(batch_size(), a_data, b_data, y.data(), &quantization_params);
239 const int8_t* a_data = inplace_a() ? y.data() : a.data(); variable
267 float(int32_t(a_data[i]) - int32_t(a_zero_point() - 0x80)) * (a_scale() / y_scale()) +
271 y_ref[i] = xnn_qs8_quantize_add(a_data[i], b_data[i], scalar_quantization_params);
275 vadd_minmax(batch_size(), a_data, b_data, y.data(), &quantization_params);
Dvaddc-microkernel-tester.h158 const int8_t* a_data = inplace() ? y.data() : a.data(); variable
186 float(int32_t(a_data[i]) - int32_t(a_zero_point() - 0x80)) * (a_scale() / y_scale()) +
190 y_ref[i] = xnn_qs8_quantize_add(a_data[i], b, scalar_quantization_params);
194 vadd_minmax(batch_size(), a_data, &b, y.data(), &quantization_params);
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dself_adjoint_eig_test.cc149 auto a_data = CreateR3Parameter<float>(batch_3d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
153 ComputeAndCompareR3<float>(&builder, batch_3d_4x4_, {a_data.get()}, in XLA_TEST_F()
161 auto a_data = CreateR3Parameter<float>( in XLA_TEST_F() local
166 ComputeAndCompareR3<float>(&builder, batch_3d_4x4_, {a_data.get()}, in XLA_TEST_F()
174 auto a_data = CreateR3Parameter<float>( in XLA_TEST_F() local
179 ComputeAndCompareR3<float>(&builder, batch_3d_4x4_, {a_data.get()}, in XLA_TEST_F()
187 auto a_data = CreateR3Parameter<float>(batch_3d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
192 {a_data.get()}, ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F()
199 auto a_data = CreateR2Parameter<float>(low_rank_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
203 ComputeAndCompareR2<float>(&builder, low_rank_4x4_, {a_data.get()}, in XLA_TEST_F()
[all …]
Dsvd_test.cc129 auto a_data = CreateR2Parameter<float>(simple_2d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
133 ComputeAndCompareR2<float>(&builder, simple_2d_4x4_, {a_data.get()}, in XLA_TEST_F()
141 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
145 ComputeAndCompareR3<float>(&builder, batch_3d_4x5_, {a_data.get()}, in XLA_TEST_F()
153 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
158 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(2, 4), {a_data.get()}, in XLA_TEST_F()
166 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
170 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(2, 5), {a_data.get()}, in XLA_TEST_F()
183 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
187 ComputeAndCompareR2<float>(&builder, singular_values, {a_data.get()}, in XLA_TEST_F()
[all …]
Dmatrix_test.cc68 auto a_data = CreateR3Parameter<int32>(input, 0, "a", &builder, &a); in XLA_TEST_F() local
73 ComputeAndCompareR3<int32>(&builder, expected, {a_data.get()}); in XLA_TEST_F()
83 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal() local
86 ComputeAndCompareR2<T>(&builder, kv.second, {a_data.get()}); in TestMatrixDiagonal()
98 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestSetMatrixDiagonal() local
106 ComputeAndCompareR2<T>(&builder, kv.second, {a_data.get(), new_diag.get()}); in TestSetMatrixDiagonal()
144 auto a_data = CreateR4Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal4D() local
147 ComputeAndCompareR3<T>(&builder, kv.second, {a_data.get()}); in TestMatrixDiagonal4D()
183 auto a_data = in XLA_TEST_F() local
195 {a_data.get(), row_data.get(), index_data.get()}); in XLA_TEST_F()
[all …]
Dlogdet_test.cc47 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
50 ComputeAndCompareR0<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
67 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
70 ComputeAndCompareR0<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
95 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
98 ComputeAndCompareR1<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
Dqr_test.cc49 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
62 ComputeAndCompareR2<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()
78 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
91 ComputeAndCompareR2<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()
116 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
123 ComputeAndCompareR3<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()
Dslicing_test.cc63 auto a_data = CreateR2Parameter<float>(BValsRight(), 0, "a", &builder, &a); in XLA_TEST_F() local
69 {a_data.get(), x_data.get(), y_data.get()}, in XLA_TEST_F()
77 auto a_data = in XLA_TEST_F() local
85 {a_data.get(), index_data.get()}); in XLA_TEST_F()
92 auto a_data = CreateR2Parameter<float>(AValsFull(), 0, "a", &builder, &a); in XLA_TEST_F() local
104 {a_data.get(), b_data.get(), x_data.get(), y_data.get()}); in XLA_TEST_F()
Dsorting_test.cc74 auto a_data = CreateR1Parameter<int>({1, 1, 2, 2, 1}, 0, "a", &builder, &a); in XLA_TEST_F() local
76 ComputeAndCompareR1<int>(&builder, {2, 3, 0, 1, 4}, {a_data.get()}); in XLA_TEST_F()
146 auto a_data = CreateR1Parameter<int>({1, 1, 2, 2, 1}, 0, "a", &builder, &a); in XLA_TEST_F() local
148 ComputeAndCompareR1<int>(&builder, {2, 3, 0}, {a_data.get()}); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtriangular_solve_test.cc107 auto a_data = in XLA_TEST_F() local
117 {a_data.get(), b_data.get()}); in XLA_TEST_F()
124 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
137 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
145 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
158 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
166 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F() local
179 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
187 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F() local
200 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
[all …]
Dcholesky_test.cc50 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
60 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
81 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
98 ComputeAndCompareR3<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
114 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
124 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
140 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
150 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
165 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
173 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
[all …]
/external/llvm-project/lldb/test/API/commands/expression/formatters/
DTestFormatters.py219 a_data = frozen.GetPointeeData()
223 a_data.GetUnsignedInt32(
239 a_data = frozen.GetPointeeData(0, 1)
242 a_data.GetUnsignedInt32(
247 a_data.GetUnsignedInt32(
252 a_data.GetUnsignedInt32(
257 a_data.GetUnsignedInt32(
262 a_data.GetUnsignedInt32(
269 a_data = frozen.GetData()
272 a_data.GetUnsignedInt32(
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils_test.cc38 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestMapNamesToNodes() local
39 test::FillIota<float>(&a_data, 1.0f); in TestMapNamesToNodes()
40 Output a_const = Const(root.WithOpName("a"), Input::Initializer(a_data)); in TestMapNamesToNodes()
71 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestMapNodesToOutputs() local
72 test::FillIota<float>(&a_data, 1.0f); in TestMapNodesToOutputs()
73 Output a_const = Const(root.WithOpName("a"), Input::Initializer(a_data)); in TestMapNodesToOutputs()
225 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestFilterGraphDef() local
226 test::FillIota<float>(&a_data, 1.0f); in TestFilterGraphDef()
227 Output a_const = Const(root.WithOpName("a"), Input::Initializer(a_data)); in TestFilterGraphDef()
285 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestGetOpTypeMatches() local
[all …]
Dfold_constants_test.cc52 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestSimpleAdd() local
53 test::FillIota<float>(&a_data, 1.0f); in TestSimpleAdd()
55 Const(root.WithOpName("a_expect_removed"), Input::Initializer(a_data)); in TestSimpleAdd()
89 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestOpExclusionAdd() local
90 test::FillIota<float>(&a_data, 1.0f); in TestOpExclusionAdd()
92 Const(root.WithOpName("a_expect_remains"), Input::Initializer(a_data)); in TestOpExclusionAdd()
300 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestRemoveUnusedNodes() local
301 test::FillIota<float>(&a_data, 1.0f); in TestRemoveUnusedNodes()
302 Output a_const = Const(root.WithOpName("a"), Input::Initializer(a_data)); in TestRemoveUnusedNodes()
338 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestMaxConstantSizeInBytes() local
[all …]
Dfile_utils_test.cc38 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestLoadTextOrBinaryGraphFile() local
39 test::FillIota<float>(&a_data, 1.0f); in TestLoadTextOrBinaryGraphFile()
40 Output a_const = Const(root.WithOpName("a"), Input::Initializer(a_data)); in TestLoadTextOrBinaryGraphFile()
Dtransform_graph_test.cc58 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestConstantFolding() local
59 test::FillIota<float>(&a_data, 1.0f); in TestConstantFolding()
61 Const(root.WithOpName("a_expect_removed"), Input::Initializer(a_data)); in TestConstantFolding()
/external/tensorflow/tensorflow/core/kernels/
Dquantized_matmul_op.cc35 void GemmlowpMultiply(OpKernelContext* op_context, const quint8* a_data, in GemmlowpMultiply() argument
38 const uint8* a_data_as_uint8 = &(a_data->value); in GemmlowpMultiply()
123 const T1* a_data = a.flat<T1>().data(); in Compute() local
141 meta::QuantizedGemm(context, transpose_a_, transpose_b_, a_data, b_data, in Compute()
151 GemmlowpMultiply<true, true, false>(context, a_data, b_data, c_data, in Compute()
155 GemmlowpMultiply<true, false, false>(context, a_data, b_data, c_data, in Compute()
161 GemmlowpMultiply<false, true, false>(context, a_data, b_data, c_data, in Compute()
165 GemmlowpMultiply<false, false, false>(context, a_data, b_data, c_data, in Compute()
172 transpose_a_, transpose_b_, transpose_c, m, n, k, a_data, offset_a, in Compute()
Dmeta_support.cc118 void QuantizedGemmImpl(OpKernelContext* tf_context, const quint8* a_data, in QuantizedGemmImpl() argument
133 params.lhs = reinterpret_cast<const uint8_t*>(&(a_data->value)); in QuantizedGemmImpl()
221 bool transpose_b, const quint8* a_data, const quint8* b_data, in QuantizedGemm() argument
230 tf_context, a_data, b_data, c_data, m, n, k, offset_a, offset_b, lda, in QuantizedGemm()
235 tf_context, a_data, b_data, c_data, m, n, k, offset_a, offset_b, lda, in QuantizedGemm()
242 tf_context, a_data, b_data, c_data, m, n, k, offset_a, offset_b, lda, in QuantizedGemm()
247 tf_context, a_data, b_data, c_data, m, n, k, offset_a, offset_b, lda, in QuantizedGemm()
Dremote_fused_graph_execute_op_test_utils.cc59 Tensor a_data(DT_FLOAT, TensorShape({1, 1, 1, 1})); in BuildMultipleAddGraph() local
60 test::FillIota<float>(&a_data, 1.0f); in BuildMultipleAddGraph()
61 Output a_const = ops::Const(root.WithOpName("A"), Input::Initializer(a_data)); in BuildMultipleAddGraph()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcholesky_thunk.cc77 se::DeviceMemoryBase a_data = in ExecuteOnStream() local
84 context->Potrf(uplo_, n_, se::DeviceMemory<float>(a_data), n_, in ExecuteOnStream()
90 uplo_, n_, se::DeviceMemory<double>(a_data), n_, info_data, in ExecuteOnStream()
96 uplo_, n_, se::DeviceMemory<std::complex<float>>(a_data), n_, in ExecuteOnStream()
102 uplo_, n_, se::DeviceMemory<std::complex<double>>(a_data), n_, in ExecuteOnStream()
Dtriangular_solve_thunk.cc91 se::DeviceMemoryBase a_data = in ExecuteOnStream() local
101 se::DeviceMemory<float>(a_data), lda, in ExecuteOnStream()
111 se::DeviceMemory<double>(a_data), lda, in ExecuteOnStream()
122 se::DeviceMemory<std::complex<float>>(a_data), in ExecuteOnStream()
133 se::DeviceMemory<std::complex<double>>(a_data), in ExecuteOnStream()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_rewriter_transform_test.cc45 Tensor a_data(DT_FLOAT, TensorShape({1, 1, 1, 1})); in TEST() local
46 test::FillIota<float>(&a_data, 1.0f); in TEST()
47 Output a_const = ops::Const(root.WithOpName("a"), Input::Initializer(a_data)); in TEST()
/external/kernel-headers/original/uapi/linux/
Da.out.h96 #define N_TRELOFF(x) (N_DATOFF(x) + (x).a_data)
142 #define N_BSSADDR(x) (N_DATADDR(x) + (x).a_data)

12