Searched refs:batched_splits_1 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_variant_op_test.cc | 105 const std::vector<int64> batched_splits_1 = {0, 2, 3, 3}; in TEST_F() local
108 BuildEncodeRaggedTensorGraph<int, int64>({batched_splits_1, batched_splits_2}, in TEST_F()
186 const std::vector<int64> batched_splits_1 = {0, 1, 2}; variable
190 BuildEncodeRaggedTensorGraph<int, int64>({batched_splits_1, batched_splits_2},
212 const std::vector<int64> batched_splits_1 = {0, 3, 3, 5, 7}; in TEST_F() local
217 BuildEncodeRaggedTensorGraph<int, int64>({batched_splits_1, batched_splits_2}, in TEST_F()
252 const std::vector<int64> batched_splits_1 = {0, 5, 10}; in TEST_F() local
261 {batched_splits_1, batched_splits_2, batched_splits_3}, TensorShape({14}), in TEST_F()
290 const std::vector<int> batched_splits_1 = {0, 5, 10}; 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 | 109 const std::vector<int64> batched_splits_1 = {0, 5}; in TEST_F() local
118 test::FillValues<int64>(&expected_splits_1, batched_splits_1); in TEST_F()
151 const std::vector<int64> batched_splits_1 = {0, 2, 4}; in TEST_F() local
168 test::FillValues<int64>(&expected_splits_1, batched_splits_1); in TEST_F()
203 const std::vector<int64> batched_splits_1 = {0, 5, 10}; in TEST_F() local
216 test::FillValues<int64>(&expected_splits_1, batched_splits_1); in TEST_F()
272 const std::vector<int64> batched_splits_1 = {0, 2, 4}; in TEST_F() local
289 test::FillValues<int64>(&expected_splits_1, batched_splits_1); in TEST_F()
323 const std::vector<int64> batched_splits_1 = {0, 2, 2}; in TEST_F() local
330 test::FillValues<int64>(&expected_splits_1, batched_splits_1); in TEST_F()
[all …]
|