Home
last modified time | relevance | path

Searched refs:batched_values (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_variant_op_test.cc133 const std::vector<int> batched_values = {1, 2, 3, 4, 5, 6}; variable
136 batched_values, true);
162 const std::vector<int> batched_values = {1, 2, 4, 5, 6, 7}; variable
165 {batched_splits}, TensorShape({3, 2}), batched_values, true);
188 const std::vector<int> batched_values = {1, 2, 4, 5, 6, 7}; variable
191 TensorShape({3, 2}), batched_values,
214 const std::vector<int> batched_values = {1, 2, 3, 4, 5, 6, 7, 8, in TEST_F() local
218 TensorShape({15}), batched_values, in TEST_F()
257 const std::vector<int> batched_values = {0, 1, 1, 2, 2, 3, 4, in TEST_F() local
262 batched_values, true); in TEST_F()
[all …]
Dragged_tensor_from_variant_op_test.cc153 const std::vector<int> batched_values = {1, 2, 3, 4, 5, 6}; in TEST_F() local
170 test::FillValues<int>(&expected_values, batched_values); in TEST_F()
208 const std::vector<int> batched_values = {0, 0, 1, 0, 1, 0, 1, 2, 0, 1, 2, in TEST_F() local
219 test::FillValues<int>(&expected_values, batched_values); in TEST_F()
280 const std::vector<int> batched_values = { in TEST_F() local
293 test::FillValues<int>(&expected_values, batched_values); in TEST_F()
325 const std::vector<int> batched_values = {1, 2, 3}; in TEST_F() local
332 test::FillValues<int>(&expected_values, batched_values); in TEST_F()
360 const std::vector<int> batched_values = {1, 2, 1, 2, 3, 4}; in TEST_F() local
367 test::FillValues<int>(&expected_values, batched_values); in TEST_F()
[all …]