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/
Dself_adjoint_eig_test.cc149 auto a_data = CreateR3Parameter<float>(batch_3d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
161 auto a_data = CreateR3Parameter<float>( in XLA_TEST_F() local
174 auto a_data = CreateR3Parameter<float>( in XLA_TEST_F() local
187 auto a_data = CreateR3Parameter<float>(batch_3d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
199 auto a_data = CreateR2Parameter<float>(low_rank_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
215 auto a_data = CreateR2Parameter<float>(matrix2d_8x8_, 0, "a", &builder, &a); in XLA_TEST_F() local
229 auto a_data = CreateR2Parameter<float>(matrix2d_8x8_, 0, "a", &builder, &a); in XLA_TEST_F() local
244 auto a_data = CreateR2Parameter<int>(wrong_type_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
255 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F() local
268 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F() local
[all …]
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 …]
Dqr_test.cc49 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
78 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
116 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
Dlogdet_test.cc47 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
67 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
95 auto a_data = CreateR3Parameter<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
83 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal() local
98 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestSetMatrixDiagonal() local
144 auto a_data = CreateR4Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal4D() local
183 auto a_data = in XLA_TEST_F() local
204 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.cc107 auto a_data = in XLA_TEST_F() local
124 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
145 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
166 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F() local
187 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F() local
208 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
230 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
252 auto a_data = in XLA_TEST_F() local
271 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
293 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
Dreduce_precision_test.cc522 std::unique_ptr<GlobalData> a_data = in DoIt() local
/external/XNNPACK/test/
Dvbinaryc-microkernel-tester.h107 const uint16_t* a_data = inplace() ? y.data() : a.data(); in Test() local
172 const uint16_t* a_data = inplace() ? y.data() : a.data(); in Test() local
254 const float* a_data = inplace() ? y.data() : a.data(); variable
318 const float* a_data = inplace() ? y.data() : a.data(); variable
406 const float* a_data = inplace() ? y.data() : a.data(); variable
Dvbinary-microkernel-tester.h115 const uint16_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
177 const uint16_t* a_data = inplace_a() ? y.data() : a.data(); in Test() local
256 const float* a_data = inplace_a() ? y.data() : a.data(); variable
317 const float* a_data = inplace_a() ? y.data() : a.data(); variable
401 const float* a_data = inplace_a() ? y.data() : a.data(); variable
Dvadd-microkernel-tester.h168 const uint8_t* a_data = inplace_a() ? y.data() : a.data(); variable
239 const int8_t* a_data = inplace_a() ? y.data() : a.data(); variable
Dvaddc-microkernel-tester.h158 const int8_t* a_data = inplace() ? y.data() : a.data(); variable
/external/gemmlowp/test/
Dtest_data.cc35 unsigned char a_data[a_count] = { variable
/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
/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/core/kernels/hexagon/
Dhexagon_rewriter_transform_test.cc45 Tensor a_data(DT_FLOAT, TensorShape({1, 1, 1, 1})); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcholesky_thunk.cc77 se::DeviceMemoryBase a_data = in ExecuteOnStream() local
Dtriangular_solve_thunk.cc91 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
/external/llvm-project/lldb/unittests/Utility/
DRangeMapTest.cpp108 bool operator()(const uint32_t a_data, const uint32_t b_data) { in TEST()

12