Searched refs:CheckAttrsExist (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 429 TF_EXPECT_OK(CheckAttrsExist(node, {})); in TEST()
430 TF_EXPECT_OK(CheckAttrsExist(node, {"apple"})); in TEST()
431 TF_EXPECT_OK(CheckAttrsExist(node, {"pear"})); in TEST()
432 TF_EXPECT_OK(CheckAttrsExist(node, {"apple", "pear"})); in TEST()
433 TF_EXPECT_OK(CheckAttrsExist(node, {"pear", "apple"})); in TEST()
442 EXPECT_FALSE(CheckAttrsExist(node, {""}).ok()); in TEST()
443 EXPECT_FALSE(CheckAttrsExist(node, {"pear", "cherry"}).ok()); in TEST()
444 EXPECT_FALSE(CheckAttrsExist(node, {"banana", "apple"}).ok()); in TEST()
|
| D | utils.h | 380 Status CheckAttrsExist(const NodeDef& node, absl::Span<const string> keys);
|
| D | utils.cc | 470 Status CheckAttrsExist(const NodeDef& node, absl::Span<const string> keys) { in CheckAttrsExist() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2313 CheckAttrsExist(*node, {"new_axis_mask", "shrink_axis_mask"})); in SimplifyStridedSlice()
2340 CheckAttrsExist(*node, {"begin_mask", "end_mask", "ellipsis_mask"})); in SimplifyStridedSlice()
|
| D | arithmetic_optimizer.cc | 2328 TF_RETURN_IF_ERROR(CheckAttrsExist(*scale, {"dtype"})); in TrySimplify()
3907 CheckAttrsExist(*node, {"begin_mask", "end_mask", "ellipsis_mask", in GetStridedSliceAxis()
|