Searched refs:CheckAttrsExist (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 426 TF_EXPECT_OK(CheckAttrsExist(node, {})); in TEST()
427 TF_EXPECT_OK(CheckAttrsExist(node, {"apple"})); in TEST()
428 TF_EXPECT_OK(CheckAttrsExist(node, {"pear"})); in TEST()
429 TF_EXPECT_OK(CheckAttrsExist(node, {"apple", "pear"})); in TEST()
430 TF_EXPECT_OK(CheckAttrsExist(node, {"pear", "apple"})); in TEST()
438 EXPECT_FALSE(CheckAttrsExist(node, {""}).ok()); in TEST()
439 EXPECT_FALSE(CheckAttrsExist(node, {"pear", "cherry"}).ok()); in TEST()
440 EXPECT_FALSE(CheckAttrsExist(node, {"banana", "apple"}).ok()); in TEST()
|
| D | utils.h | 366 Status CheckAttrsExist(const NodeDef& node, absl::Span<const string> keys);
|
| D | utils.cc | 485 Status CheckAttrsExist(const NodeDef& node, absl::Span<const string> keys) { in CheckAttrsExist() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 2235 TF_RETURN_IF_ERROR(CheckAttrsExist(*scale, {"dtype"})); in TrySimplify()
3323 CheckAttrsExist(*node, {"begin_mask", "end_mask", "ellipsis_mask", in GetStridedSliceAxis()
|
| D | constant_folding.cc | 2265 CheckAttrsExist(*node, {"new_axis_mask", "shrink_axis_mask"})); in SimplifyStridedSlice()
2292 CheckAttrsExist(*node, {"begin_mask", "end_mask", "ellipsis_mask"})); in SimplifyStridedSlice()
|