Searched refs:CheckAttrExists (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 420 TEST(CheckAttrExists, All) { in TEST() argument
426 TF_EXPECT_OK(CheckAttrExists(node, "apple")); in TEST()
427 TF_EXPECT_OK(CheckAttrExists(node, "pear")); in TEST()
435 Status status = CheckAttrExists(node, "banana"); in TEST()
|
| D | utils.cc | 462 Status CheckAttrExists(const NodeDef& node, const string& key) { in CheckAttrExists() function
472 TF_RETURN_IF_ERROR(CheckAttrExists(node, key)); in CheckAttrsExist()
|
| D | utils.h | 377 Status CheckAttrExists(const NodeDef& node, const string& key);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 344 CheckAttrExists(*node, "value").ok() && in GetTensorFromConstNode()
1664 TF_RETURN_IF_ERROR(CheckAttrExists(node, "N")); in InitializeChains()
2329 TF_RETURN_IF_ERROR(CheckAttrExists(*weights, "dtype")); in TrySimplify()
3612 TF_RETURN_IF_ERROR(CheckAttrExists(*root, "T")); in TrySimplify()
3773 TF_RETURN_IF_ERROR(CheckAttrExists(*pack, "axis")); in CheckInputs()
3825 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_begin, "value")); in GetSimpleSliceAxis()
3829 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_size, "value")); in GetSimpleSliceAxis()
3937 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_begin, "value")); in GetStridedSliceAxis()
3941 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_end, "value")); in GetStridedSliceAxis()
3945 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_strides, "value")); in GetStridedSliceAxis()
|
| D | constant_folding.cc | 388 CheckAttrExists(*node, "value").ok() && in GetTensorFromConstNode()
684 TF_RETURN_IF_ERROR(CheckAttrExists(node, "T")); in MaterializeBroadcastGradientArgs()
856 TF_RETURN_IF_ERROR(CheckAttrExists(*input_node, "value")); in MaterializeConstantValuedNode()
1362 TF_RETURN_IF_ERROR(CheckAttrExists(*input_node, "value")); in EvaluateOneFoldable()
2524 !CheckAttrExists(*output_idx_node, "value").ok()) { in SimplifyCase()
3023 TF_RETURN_IF_ERROR(CheckAttrExists(*node, "T")); in SimplifyArithmeticOperations()
3183 if (!CheckAttrExists(parent, "T").ok()) return false; in PrepareConstantPushDown()
3391 if (!CheckAttrExists(*node, "T").ok()) return false; in ConstantPushDown()
|