Home
last modified time | relevance | path

Searched defs:a_data (Results 1 – 25 of 100) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dtriangular_solve_test.cc108 auto a_data = in XLA_TEST_F() local
125 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
146 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
167 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F() local
188 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F() local
209 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
231 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
253 auto a_data = in XLA_TEST_F() local
272 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
294 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F() local
[all …]
H A Dcholesky_test.cc50 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
81 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
114 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
140 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
165 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
196 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dsvd_test.cc129 auto a_data = CreateR2Parameter<float>(simple_2d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
141 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
153 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
166 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
183 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
197 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F() local
209 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F() local
221 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F() local
235 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F() local
249 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F() local
[all …]
H A Dself_adjoint_eig_test.cc139 auto a_data = CreateR3Parameter<float>(batch_3d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
157 auto a_data = CreateParameter<complex64>(input, 0, "a", &builder, &a); in XLA_TEST_F() local
169 auto a_data = CreateR3Parameter<float>( in XLA_TEST_F() local
182 auto a_data = CreateR3Parameter<float>( in XLA_TEST_F() local
195 auto a_data = CreateR3Parameter<float>(batch_3d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
207 auto a_data = CreateR2Parameter<float>(low_rank_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
223 auto a_data = CreateR2Parameter<float>(matrix2d_8x8_, 0, "a", &builder, &a); in XLA_TEST_F() local
237 auto a_data = CreateR2Parameter<float>(matrix2d_8x8_, 0, "a", &builder, &a); in XLA_TEST_F() local
252 auto a_data = CreateR2Parameter<int>(wrong_type_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
280 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_P() local
H A Dmatrix_test.cc68 auto a_data = CreateR3Parameter<int32_t>(input, 0, "a", &builder, &a); in XLA_TEST_F() local
87 auto a_data = CreateParameter<float>(input, 0, "a", &builder, &a); in XLA_TEST_F() local
111 auto a_data = CreateParameter<complex64>(input, 0, "a", &builder, &a); in XLA_TEST_F() local
135 auto a_data = CreateParameter<float>(input, 0, "a", &builder, &a); in XLA_TEST_F() local
159 auto a_data = CreateParameter<complex64>(input, 0, "a", &builder, &a); in XLA_TEST_F() local
179 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal() local
194 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestSetMatrixDiagonal() local
240 auto a_data = CreateR4Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal4D() local
279 auto a_data = in XLA_TEST_F() local
300 auto a_data = in XLA_TEST_F() local
H A Dqr_test.cc55 auto a_data = CreateParameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
93 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
129 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
151 auto a_data = CreateParameter<xla::complex64>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
H A Dlogdet_test.cc45 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
68 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
102 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
148 auto a_data = CreateParameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
H A Dslicing_test.cc63 auto a_data = CreateR2Parameter<float>(BValsRight(), 0, "a", &builder, &a); in XLA_TEST_F() local
77 auto a_data = in XLA_TEST_F() local
92 auto a_data = in XLA_TEST_F() local
109 auto a_data = CreateR2Parameter<float>(AValsFull(), 0, "a", &builder, &a); in XLA_TEST_F() local
128 auto a_data = CreateR2Parameter<float>(AValsFull(), 0, "a", &builder, &a); in XLA_TEST_F() local
H A Dsorting_test.cc74 auto a_data = CreateR1Parameter<int>({1, 1, 2, 2, 1}, 0, "a", &builder, &a); in XLA_TEST_F() local
146 auto a_data = CreateR1Parameter<int>({1, 1, 2, 2, 1}, 0, "a", &builder, &a); in XLA_TEST_F() local
/aosp_15_r20/external/executorch/kernels/portable/test/
H A Dop_allclose_test.cpp271 auto a_data = a.data_ptr<float>(); in TEST() local
293 auto a_data = a.data_ptr<double>(); in TEST() local
314 auto a_data = a.data_ptr<float>(); in TEST() local
335 auto a_data = a.data_ptr<double>(); in TEST() local
356 auto a_data = a.data_ptr<float>(); in TEST() local
377 auto a_data = a.data_ptr<double>(); in TEST() local
398 auto a_data = a.data_ptr<float>(); in TEST() local
419 auto a_data = a.data_ptr<double>(); in TEST() local
437 auto a_data = a.data_ptr<float>(); in TEST() local
461 auto a_data = a.data_ptr<double>(); in TEST() local
H A Dop_mul_test.cpp46 std::vector<exec_aten::qint8> a_data{}; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dtransform_utils_test.cc38 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestMapNamesToNodes() local
71 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestMapNodesToOutputs() local
225 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestFilterGraphDef() local
285 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestGetOpTypeMatches() local
382 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestGetOpTypeMatchesDAG() local
415 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestReplaceMatchingOpTypes() local
524 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestRenameNodeInputs() local
558 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestRenameNodeInputsWithRedirects() local
596 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestRenameNodeInputsWithCycle() local
630 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestRenameNodeInputsWithWildcard() local
[all …]
H A Dfold_constants_test.cc52 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestSimpleAdd() local
89 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestOpExclusionAdd() local
300 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestRemoveUnusedNodes() local
338 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestMaxConstantSizeInBytes() local
/aosp_15_r20/external/XNNPACK/test/
H A Dvbinaryc-microkernel-tester.h101 const uint16_t* a_data = inplace() ? y.data() : a.data(); in Test() local
165 const uint16_t* a_data = inplace() ? y.data() : a.data(); in Test() local
247 const float* a_data = inplace() ? y.data() : a.data(); variable
318 const float* a_data = inplace() ? y.data() : a.data(); in Test() local
388 const float* a_data = inplace() ? y.data() : a.data(); in Test() local
H A Dvbinary-microkernel-tester.h109 const uint16_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
170 const uint16_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
249 const float* a_data = inplace_a() ? y.data() : a.data(); variable
316 const float* a_data = inplace_a() ? y.data() : a.data(); in Test() local
382 const float* a_data = inplace_a() ? y.data() : a.data(); in Test() local
H A Dvaddc-microkernel-tester.h152 const uint8_t* a_data = inplace() ? y.data() : a.data(); in Test() local
213 const int8_t* a_data = inplace() ? y.data() : a.data(); in Test() local
H A Dvadd-microkernel-tester.h163 const uint8_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
226 const int8_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
H A Dvmulc-microkernel-tester.h154 const uint8_t* a_data = inplace() ? y.data() : a.data(); in Test() local
216 const int8_t* a_data = inplace() ? y.data() : a.data(); in Test() local
H A Dvmul-microkernel-tester.h164 const uint8_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
228 const int8_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
/aosp_15_r20/external/executorch/kernels/optimized/cpu/
H A Dop_le.cpp68 const CTYPE_A* a_data = a.const_data_ptr<CTYPE_A>(); in opt_le_tensor_out() local
138 const CTYPE_A* a_data = a.const_data_ptr<CTYPE_A>(); in opt_le_scalar_out() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_matmul_op.cc38 void GemmlowpMultiply(OpKernelContext* op_context, const quint8* a_data, in GemmlowpMultiply()
138 const T1* a_data = a.flat<T1>().data(); in Compute() local
H A Dmeta_support.cc118 void QuantizedGemmImpl(OpKernelContext* tf_context, const quint8* a_data, in QuantizedGemmImpl()
221 bool transpose_b, const quint8* a_data, const quint8* b_data, in QuantizedGemm()
/aosp_15_r20/external/executorch/backends/cadence/hifi/operators/
H A Dop_div.cpp94 float* a_data = a.mutable_data_ptr<float>(); in div_out() local
215 float* a_data = a.mutable_data_ptr<float>(); in div_out_mode() local
/aosp_15_r20/external/gemmlowp/test/
H A Dtest.cc739 const std::uint8_t a_data[] = {0, 0, 0, 0, 0, 0, 0, 0, in TestWithSmallDataPerChannelQuantization() local
809 const std::uint8_t a_data[] = { in TestWithLargeDataPerChannelQuantization() local
1025 std::vector<std::uint8_t> a_data(m * k, 0); in TestMultithreadedPerChannelQuantization() local
1117 const std::uint8_t a_data[] = {7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18}; in TestWithSmallData() local
H A Dtest_data.cc35 unsigned char a_data[a_count] = { variable

1234