Searched refs:input_suffix (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/rnn/ops/ |
| D | lstm_ops_test.cc | 52 string input_suffix = strings::StrCat(";", JoinedCopies("?", 6)); in TEST_F() local
55 INFER_ERROR("must be rank 2", op, "[?];?" + input_suffix); in TEST_F()
56 INFER_ERROR("must be rank 2", op, "?;[?]" + input_suffix); in TEST_F()
59 INFER_OK(op, "?;?" + input_suffix, JoinedCopies("[?,?]", 7)); in TEST_F()
60 INFER_OK(op, "[?,?];[?,?]" + input_suffix, JoinedCopies("[d0_0,d1_1]", 7)); in TEST_F()
67 string input_suffix = strings::StrCat(";", JoinedCopies("?", 14)); in TEST_F() local
70 INFER_ERROR("must be rank 2", op, "[?];?" + input_suffix); in TEST_F()
71 INFER_ERROR("must be rank 2", op, "?;[?]" + input_suffix); in TEST_F()
74 INFER_OK(op, "?;?" + input_suffix, "[?,?];[?,?];[?];[?];[?]"); in TEST_F()
75 INFER_OK(op, "[?,?];[?,?]" + input_suffix, in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1582 ShapeHandle input_suffix; in ScatterNdUpdateShape() local
1583 TF_RETURN_IF_ERROR(c->Subshape(input_shape, ix, &input_suffix)); in ScatterNdUpdateShape()
1587 s = c->Merge(input_suffix, suffix_updates, &unused); in ScatterNdUpdateShape()
|