Home
last modified time | relevance | path

Searched refs:CreateTestData (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/federated-compute/fcp/aggregation/core/
H A Done_dim_grouping_aggregator_test.cc137 Tensor::Create(DT_INT64, {}, CreateTestData<int64_t>({0})).value(); in TEST()
138 Tensor t1 = Tensor::Create(DT_INT32, {}, CreateTestData({1})).value(); in TEST()
139 Tensor t2 = Tensor::Create(DT_INT32, {}, CreateTestData({2})).value(); in TEST()
140 Tensor t3 = Tensor::Create(DT_INT32, {}, CreateTestData({3})).value(); in TEST()
157 Tensor::Create(DT_INT64, shape, CreateTestData<int64_t>({0, 1, 2, 3})) in TEST()
160 Tensor::Create(DT_INT32, shape, CreateTestData({1, 3, 15, 27})).value(); in TEST()
162 Tensor::Create(DT_INT32, shape, CreateTestData({10, 5, 1, 2})).value(); in TEST()
164 Tensor::Create(DT_INT32, shape, CreateTestData({3, 11, 7, 20})).value(); in TEST()
184 Tensor::Create(DT_INT64, shape, CreateTestData<int64_t>({3, 3, 2, 0})) in TEST()
187 Tensor::Create(DT_INT32, shape, CreateTestData({1, 3, 15, 27})).value(); in TEST()
[all …]
H A Dcomposite_key_combiner_test.cc51 Tensor::Create(DT_FLOAT, {3}, CreateTestData<float>({1.1, 1.2, 1.3})) in TEST()
54 Tensor::Create(DT_INT32, {4}, CreateTestData<int32_t>({1, 2, 3, 4})) in TEST()
63 Tensor::Create(DT_FLOAT, {3}, CreateTestData<float>({1.1, 1.2, 1.3})) in TEST()
66 Tensor::Create(DT_INT32, {3}, CreateTestData<int32_t>({1, 2, 3})).value(); in TEST()
74 Tensor::Create(DT_FLOAT, {3}, CreateTestData<float>({1.1, 1.2, 1.3})) in TEST()
77 Tensor::Create(DT_INT32, {3}, CreateTestData<int32_t>({1, 2, 3})).value(); in TEST()
79 Tensor::Create(DT_INT32, {3}, CreateTestData<int32_t>({4, 5, 6})).value(); in TEST()
88 Tensor::Create(DT_FLOAT, {3}, CreateTestData<float>({1.1, 1.2, 1.3})) in TEST()
91 Tensor::Create(DT_INT32, {3}, CreateTestData<int32_t>({1, 2, 3})).value(); in TEST()
106 Tensor::Create(DT_FLOAT, {1}, CreateTestData<float>({1.3})).value(); in TEST()
[all …]
H A Dagg_vector_aggregator_test.cc61 Tensor t1 = Tensor::Create(DT_INT32, {}, CreateTestData({1})).value(); in TEST()
62 Tensor t2 = Tensor::Create(DT_INT32, {}, CreateTestData({2})).value(); in TEST()
63 Tensor t3 = Tensor::Create(DT_INT32, {}, CreateTestData({3})).value(); in TEST()
80 Tensor::Create(DT_INT32, shape, CreateTestData({1, 3, 15, 27})).value(); in TEST()
82 Tensor::Create(DT_INT32, shape, CreateTestData({10, 5, 1, 2})).value(); in TEST()
84 Tensor::Create(DT_INT32, shape, CreateTestData({3, 11, 7, 20})).value(); in TEST()
103 Tensor t1 = Tensor::Create(DT_INT32, {}, CreateTestData({1})).value(); in TEST()
104 Tensor t2 = Tensor::Create(DT_INT32, {}, CreateTestData({2})).value(); in TEST()
105 Tensor t3 = Tensor::Create(DT_INT32, {}, CreateTestData({3})).value(); in TEST()
122 Tensor t = Tensor::Create(DT_FLOAT, {}, CreateTestData<float>({0})).value(); in TEST()
[all …]
H A Dfederated_sum_test.cc40 Tensor t1 = Tensor::Create(DT_INT32, {}, CreateTestData({1})).value(); in TEST()
41 Tensor t2 = Tensor::Create(DT_INT32, {}, CreateTestData({2})).value(); in TEST()
42 Tensor t3 = Tensor::Create(DT_INT32, {}, CreateTestData({3})).value(); in TEST()
61 Tensor::Create(DT_INT32, shape, CreateTestData({1, 3, 15, 27})).value(); in TEST()
63 Tensor::Create(DT_INT32, shape, CreateTestData({10, 5, 1, 2})).value(); in TEST()
65 Tensor::Create(DT_INT32, shape, CreateTestData({3, 11, 7, 20})).value(); in TEST()
87 Tensor t1 = Tensor::Create(DT_INT32, {}, CreateTestData({1})).value(); in TEST()
88 Tensor t2 = Tensor::Create(DT_INT32, {}, CreateTestData({2})).value(); in TEST()
89 Tensor t3 = Tensor::Create(DT_INT32, {}, CreateTestData({3})).value(); in TEST()
H A Dagg_vector_test.cc38 auto t1 = Tensor::Create(DT_INT32, {}, CreateTestData({0})); in TEST()
41 auto t2 = Tensor::Create(DT_FLOAT, {3}, CreateTestData<float>({0, 1, 2})); in TEST()
46 auto t = Tensor::Create(DT_INT32, {}, CreateTestData({5})); in TEST()
51 auto t = Tensor::Create(DT_INT32, {2}, CreateTestData({3, 14})); in TEST()
57 auto t = Tensor::Create(DT_FLOAT, {4}, CreateTestData<float>({2, 3, 4, 5})); in TEST()
68 auto t = Tensor::Create(DT_FLOAT, {4}, CreateTestData<float>({2, 3, 4, 5})); in TEST()
H A Dtensor_test.cc41 auto t = Tensor::Create(DT_FLOAT, {3}, CreateTestData<float>({1, 2, 3})); in TEST()
51 CreateTestData<string_view>({"foo", "bar"})); in TEST()
60 auto t = Tensor::Create(DT_FLOAT, {}, CreateTestData<char>({'a', 'b', 'c'})); in TEST()
65 auto t = Tensor::Create(DT_FLOAT, {1}, CreateTestData<float>({1, 2})); in TEST()
72 auto t = Tensor::Create(DT_FLOAT, {1}, CreateTestData<float>({1})); in TEST()
104 auto t = Tensor::Create(DT_INT32, {2, 2}, CreateTestData(values)); in TEST()
116 auto t = Tensor::Create(DT_STRING, {2, 3}, CreateTestData(values)); in TEST()
154 Tensor::Create(DT_STRING, {3}, CreateTestData<string_view>({s1, s2, s3})); in TEST()
H A Dinput_tensor_list_test.cc39 : t1_(Tensor::Create(DT_FLOAT, {1}, CreateTestData<float>({1})).value()), in InputTensorListTest()
40 t2_(Tensor::Create(DT_INT32, {2}, CreateTestData<int32_t>({2, 3}))
42 t3_(Tensor::Create(DT_INT64, {3}, CreateTestData<int64_t>({4, 5, 6}))
44 t4_(Tensor::Create(DT_FLOAT, {4}, CreateTestData<float>({7, 8, 9, 10}))
47 CreateTestData<int32_t>({11, 12, 13, 14, 15}))
50 CreateTestData<int64_t>({16, 17, 18, 19, 20, 21}))
/aosp_15_r20/external/federated-compute/fcp/aggregation/protocol/simple_aggregation/
H A Dsimple_aggregation_protocol_test.cc440 return Tensor::Create(DT_INT32, {}, CreateTestData({1})); in TEST_F()
519 return Tensor::Create(DT_FLOAT, {}, CreateTestData({2.f})); in TEST_F()
540 return Tensor::Create(DT_INT32, {}, CreateTestData({1})); in TEST_F()
700 CreateTestData({4, 3, 11, 7, 1, 6})); in TEST_F()
703 return Tensor::Create(DT_FLOAT, {}, CreateTestData({1.f})); in TEST_F()
709 CreateTestData({1, 8, 2, 10, 13, 2})); in TEST_F()
712 return Tensor::Create(DT_FLOAT, {}, CreateTestData({2.f})); in TEST_F()
781 return Tensor::Create(DT_INT32, {}, CreateTestData({1})); in TEST_F()
818 return Tensor::Create(DT_INT32, {}, CreateTestData({++tensor_value})); in TEST_F()
924 return Tensor::Create(DT_INT32, {}, CreateTestData({1})); in TEST_F()
/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/
H A Dtensorflow_checkpoint_builder_factory_test.cc49 DT_FLOAT, TensorShape({4}), CreateTestData<float>({1.0, 2.0, 3.0, 4.0})); in TEST()
52 CreateTestData<float>({5.0, 6.0})); in TEST()
70 DT_FLOAT, TensorShape({4}), CreateTestData<float>({1.0, 2.0, 3.0, 4.0})); in TEST()
73 CreateTestData<float>({5.0, 6.0})); in TEST()
H A Dcheckpoint_writer_test.cc44 CreateTestData<float>({1.0, 2.0, 3.0, 4.0})) in TEST()
47 CreateTestData<int32_t>({11, 12, 13, 14, 15, 16})) in TEST()
52 CreateTestData<string_view>({"foo", "bar", "bazzzzzzzzzzzzzzzzzzz"})) in TEST()
/aosp_15_r20/external/federated-compute/fcp/aggregation/testing/
H A Dtest_data.h29 std::unique_ptr<MutableVectorData<T>> CreateTestData( in CreateTestData() function