Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dtriangular_solve_test.cc105 auto a_data = in XLA_TEST_F() local
115 {a_data.get(), b_data.get()}); in XLA_TEST_F()
122 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
135 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
143 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F() local
156 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
164 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F() local
177 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
185 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F() local
198 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
[all …]
Dreduce_precision_test.cc233 std::unique_ptr<GlobalData> a_data = in XLA_TEST_P() local
239 ComputeAndCompareR1<float>(&builder, expected_values, {a_data.get()}); in XLA_TEST_P()
256 std::unique_ptr<GlobalData> a_data = in XLA_TEST_F() local
276 ComputeAndCompareR1<float>(&builder, {0.0f}, {a_data.get()}); in XLA_TEST_F()
285 std::unique_ptr<GlobalData> a_data = in XLA_TEST_F() local
302 ComputeAndCompareR1<float>(&builder, {-1.00001f, -1.00001f}, {a_data.get()}); in XLA_TEST_F()
311 std::unique_ptr<GlobalData> a_data = in XLA_TEST_F() local
326 ComputeAndCompareR1<float>(&builder, {-1.0f, -1.0f}, {a_data.get()}); in XLA_TEST_F()
335 std::unique_ptr<GlobalData> a_data = in XLA_TEST_F() local
351 ComputeAndCompareR1<float>(&builder, {-1.00001f}, {a_data.get()}); in XLA_TEST_F()
[all …]
Dcholesky_test.cc51 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
61 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
77 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
87 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
102 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
110 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
133 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
149 ComputeAndCompareR3<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
/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.cc144 auto a_data = CreateR2Parameter<float>(simple_2d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F() local
148 ComputeAndCompareR2<float>(&builder, simple_2d_4x4_, {a_data.get()}, in XLA_TEST_F()
156 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
160 ComputeAndCompareR3<float>(&builder, batch_3d_4x5_, {a_data.get()}, in XLA_TEST_F()
168 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
173 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(2, 4), {a_data.get()}, in XLA_TEST_F()
181 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
185 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(2, 5), {a_data.get()}, in XLA_TEST_F()
198 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F() local
202 ComputeAndCompareR2<float>(&builder, singular_values, {a_data.get()}, in XLA_TEST_F()
[all …]
Dmatrix_test.cc43 auto a_data = CreateR3Parameter<int32>(input, 0, "a", &builder, &a); in XLA_TEST_F() local
48 ComputeAndCompareR3<int32>(&builder, expected, {a_data.get()}); in XLA_TEST_F()
69 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal() local
72 ComputeAndCompareR2<T>(&builder, kv.second, {a_data.get()}); in TestMatrixDiagonal()
103 auto a_data = in XLA_TEST_F() local
115 {a_data.get(), row_data.get(), index_data.get()}); in XLA_TEST_F()
124 auto a_data = in XLA_TEST_F() local
136 {a_data.get(), row_data.get(), index_data.get()}); in XLA_TEST_F()
Dslicing_test.cc61 auto a_data = CreateR2Parameter<float>(BValsRight(), 0, "a", &builder, &a); in XLA_TEST_F() local
67 {a_data.get(), x_data.get(), y_data.get()}, in XLA_TEST_F()
75 auto a_data = in XLA_TEST_F() local
83 {a_data.get(), index_data.get()}); in XLA_TEST_F()
90 auto a_data = CreateR2Parameter<float>(AValsFull(), 0, "a", &builder, &a); in XLA_TEST_F() local
102 {a_data.get(), b_data.get(), x_data.get(), y_data.get()}); in XLA_TEST_F()
Dqr_test.cc46 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
59 ComputeAndCompareR2<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()
82 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F() local
89 ComputeAndCompareR3<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()
Dsorting_test.cc87 auto a_data = CreateR1Parameter<int>({1, 1, 2, 2, 1}, 0, "a", &builder, &a); in XLA_TEST_F() local
89 ComputeAndCompareR1<int>(&builder, {2, 3, 0, 1, 4}, {a_data.get()}); in XLA_TEST_F()
/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()
Dmeta_support.h71 const quint8* a_data, const quint8* b_data, qint32* c_data,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcholesky_thunk.cc81 se::DeviceMemoryBase a_data = in ExecuteOnStream() local
88 context->Potrf(uplo_, n_, se::DeviceMemory<float>(a_data), n_, in ExecuteOnStream()
94 uplo_, n_, se::DeviceMemory<double>(a_data), n_, info_data, in ExecuteOnStream()
100 uplo_, n_, se::DeviceMemory<std::complex<float>>(a_data), n_, in ExecuteOnStream()
106 uplo_, n_, se::DeviceMemory<std::complex<double>>(a_data), n_, in ExecuteOnStream()
Dtriangular_solve_thunk.cc90 se::DeviceMemoryBase a_data = in ExecuteOnStream() local
100 se::DeviceMemory<float>(a_data), lda, in ExecuteOnStream()
110 se::DeviceMemory<double>(a_data), lda, in ExecuteOnStream()
121 se::DeviceMemory<std::complex<float>>(a_data), in ExecuteOnStream()
132 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)
/external/gemmlowp/test/
Dtest.cc739 const std::uint8_t a_data[] = {0, 0, 0, 0, 0, 0, 0, 0, in TestWithSmallDataPerChannelQuantization() local
744 MatrixMap<const std::uint8_t, MapOrder::RowMajor> lhs(a_data, m, k, lda); in TestWithSmallDataPerChannelQuantization()
809 const std::uint8_t a_data[] = { in TestWithLargeDataPerChannelQuantization() local
854 MatrixMap<const std::uint8_t, MapOrder::RowMajor> lhs(a_data, m, k, lda); in TestWithLargeDataPerChannelQuantization()
1025 std::vector<std::uint8_t> a_data(m * k, 0); in TestMultithreadedPerChannelQuantization() local
1027 a_data[i * k] = lhs_first_el[i / 16]; in TestMultithreadedPerChannelQuantization()
1028 a_data[i * k + 1] = 255; in TestMultithreadedPerChannelQuantization()
1030 a_data[i * k + j] = std::uint8_t(-lhs_offsets_terse[i / 16]); in TestMultithreadedPerChannelQuantization()
1041 MatrixMap<const std::uint8_t, MapOrder::RowMajor> lhs(&a_data[0], m, k, lda); in TestMultithreadedPerChannelQuantization()
1117 const std::uint8_t a_data[] = {7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18}; in TestWithSmallData() local
[all …]
Dtest_data.h37 extern unsigned char a_data[];
Dtest_data.cc35 unsigned char a_data[a_count] = { variable
/external/kernel-headers/original/uapi/asm-x86/asm/
Da.out.h9 unsigned a_data; /* length of data, in bytes */ member

12