Searched refs:batched_splits_2 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_variant_op_test.cc | 106 const std::vector<int64> batched_splits_2 = {0, 0, 0, 0}; in TEST_F() local
108 BuildEncodeRaggedTensorGraph<int, int64>({batched_splits_1, batched_splits_2}, in TEST_F()
187 const std::vector<int64> batched_splits_2 = {0, 2, 3}; variable
190 BuildEncodeRaggedTensorGraph<int, int64>({batched_splits_1, batched_splits_2},
213 const std::vector<int64> batched_splits_2 = {0, 1, 3, 3, 8, 11, 11, 15}; in TEST_F() local
217 BuildEncodeRaggedTensorGraph<int, int64>({batched_splits_1, batched_splits_2}, in TEST_F()
253 const std::vector<int64> batched_splits_2 = {0, 1, 3, 4, 5, 6, in TEST_F() local
261 {batched_splits_1, batched_splits_2, batched_splits_3}, TensorShape({14}), in TEST_F()
291 const std::vector<int> batched_splits_2 = {0, 1, 3, 4, 5, 6, 7, 8, 9, 10, 11}; in TEST_F() local
298 {batched_splits_1, batched_splits_2, batched_splits_3}, TensorShape({14}), in TEST_F()
[all …]
|
| D | ragged_tensor_from_variant_op_test.cc | 152 const std::vector<int64> batched_splits_2 = {0, 3, 3, 5, 6}; in TEST_F() local
169 test::FillValues<int64>(&expected_splits_2, batched_splits_2); in TEST_F()
204 const std::vector<int64> batched_splits_2 = {0, 1, 3, 4, 6, 8, in TEST_F() local
217 test::FillValues<int64>(&expected_splits_2, batched_splits_2); in TEST_F()
273 const std::vector<int64> batched_splits_2 = {0, 5, 10, 15, 20}; in TEST_F() local
290 test::FillValues<int64>(&expected_splits_2, batched_splits_2); in TEST_F()
324 const std::vector<int64> batched_splits_2 = {0, 3, 3}; in TEST_F() local
331 test::FillValues<int64>(&expected_splits_2, batched_splits_2); in TEST_F()
359 const std::vector<int64> batched_splits_2 = {0, 1, 2, 3}; in TEST_F() local
366 test::FillValues<int64>(&expected_splits_2, batched_splits_2); in TEST_F()
[all …]
|