Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/client/lib/
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 …]
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
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
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
Dmatrix_test.cc68 auto a_data = CreateR3Parameter<int32>(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
131 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal() local
146 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestSetMatrixDiagonal() local
192 auto a_data = CreateR4Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal4D() local
231 auto a_data = in XLA_TEST_F() local
252 auto a_data = in XLA_TEST_F() local
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 = CreateR2Parameter<float>(AValsFull(), 0, "a", &builder, &a); in XLA_TEST_F() local
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
/external/tensorflow/tensorflow/compiler/xla/tests/
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 …]
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
/external/XNNPACK/test/
Dvbinaryc-microkernel-tester.h102 const uint16_t* a_data = inplace() ? y.data() : a.data(); in Test() local
167 const uint16_t* a_data = inplace() ? y.data() : a.data(); in Test() local
249 const float* a_data = inplace() ? y.data() : a.data(); variable
320 const float* a_data = inplace() ? y.data() : a.data(); in Test() local
390 const float* a_data = inplace() ? y.data() : a.data(); in Test() local
Dvbinary-microkernel-tester.h110 const uint16_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
172 const uint16_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
251 const float* a_data = inplace_a() ? y.data() : a.data(); variable
318 const float* a_data = inplace_a() ? y.data() : a.data(); in Test() local
384 const float* a_data = inplace_a() ? y.data() : a.data(); in Test() local
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
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
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
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
/external/tensorflow/tensorflow/tools/graph_transforms/
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 …]
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
Dfile_utils_test.cc38 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestLoadTextOrBinaryGraphFile() local
Dtransform_graph_test.cc58 Tensor a_data(DT_FLOAT, TensorShape({width})); in TestConstantFolding() local
/external/gemmlowp/test/
Dtest_data.cc35 unsigned char a_data[a_count] = { variable
/external/tensorflow/tensorflow/core/kernels/
Dquantized_matmul_op.cc35 void GemmlowpMultiply(OpKernelContext* op_context, const quint8* a_data, in GemmlowpMultiply()
123 const T1* a_data = a.flat<T1>().data(); in Compute() local
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()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcholesky_thunk.cc81 se::DeviceMemoryBase a_data = in ExecuteOnStream() local
Dtriangular_solve_thunk.cc90 se::DeviceMemoryBase a_data = in ExecuteOnStream() local
/external/kernel-headers/original/uapi/asm-x86/asm/
Da.out.h9 unsigned a_data; /* length of data, in bytes */ member

12