Searched refs:size_splits2 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 3107 Output size_splits2 = ops::Const(s.WithOpName("size_splits2"), {20, 12}, {2}); in TEST_F() local
3108 ops::SplitV split2(s.WithOpName("split2"), x, size_splits2, axis, 2); in TEST_F()
|
| D | constant_folding_test.cc | 1798 auto size_splits2 = ops::Const(scope.WithOpName("size_splits2"), {2, 3}, {2}); in TEST_F() local
1800 ops::SplitV s2(scope.WithOpName("s2"), in2, size_splits2, split_dim, 2); in TEST_F()
|