Searched refs:CheckAttrsExist (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 372 TF_EXPECT_OK(CheckAttrsExist(node, {})); in TEST()
373 TF_EXPECT_OK(CheckAttrsExist(node, {"apple"})); in TEST()
374 TF_EXPECT_OK(CheckAttrsExist(node, {"pear"})); in TEST()
375 TF_EXPECT_OK(CheckAttrsExist(node, {"apple", "pear"})); in TEST()
376 TF_EXPECT_OK(CheckAttrsExist(node, {"pear", "apple"})); in TEST()
384 EXPECT_FALSE(CheckAttrsExist(node, {""}).ok()); in TEST()
385 EXPECT_FALSE(CheckAttrsExist(node, {"pear", "cherry"}).ok()); in TEST()
386 EXPECT_FALSE(CheckAttrsExist(node, {"banana", "apple"}).ok()); in TEST()
|
| D | utils.h | 272 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 | constant_folding.cc | 2104 CheckAttrsExist(*node, {"new_axis_mask", "shrink_axis_mask"})); in SimplifyStridedSlice()
2140 CheckAttrsExist(*node, {"begin_mask", "end_mask", "ellipsis_mask"})); in SimplifyStridedSlice()
|
| D | arithmetic_optimizer.cc | 2118 TF_RETURN_IF_ERROR(CheckAttrsExist(*scale, {"dtype", "value"})); in TrySimplify()
3139 CheckAttrsExist(*node, {"begin_mask", "end_mask", "ellipsis_mask", in GetSliceAxis()
|