Searched refs:CheckAttrExists (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 417 TEST(CheckAttrExists, All) { in TEST() argument
423 TF_EXPECT_OK(CheckAttrExists(node, "apple")); in TEST()
424 TF_EXPECT_OK(CheckAttrExists(node, "pear")); in TEST()
432 Status status = CheckAttrExists(node, "banana"); in TEST()
|
| D | utils.cc | 477 Status CheckAttrExists(const NodeDef& node, const string& key) { in CheckAttrExists() function
487 TF_RETURN_IF_ERROR(CheckAttrExists(node, key)); in CheckAttrsExist()
|
| D | utils.h | 363 Status CheckAttrExists(const NodeDef& node, const string& key);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 282 CheckAttrExists(*node, "value").ok() && in GetTensorFromConstNode()
1602 TF_RETURN_IF_ERROR(CheckAttrExists(node, "N")); in InitializeChains()
2236 TF_RETURN_IF_ERROR(CheckAttrExists(*weights, "dtype")); in TrySimplify()
3028 TF_RETURN_IF_ERROR(CheckAttrExists(*root, "T")); in TrySimplify()
3189 TF_RETURN_IF_ERROR(CheckAttrExists(*pack, "axis")); in CheckInputs()
3241 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_begin, "value")); in GetSimpleSliceAxis()
3245 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_size, "value")); in GetSimpleSliceAxis()
3353 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_begin, "value")); in GetStridedSliceAxis()
3357 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_end, "value")); in GetStridedSliceAxis()
3361 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_strides, "value")); in GetStridedSliceAxis()
|
| D | constant_folding.cc | 384 CheckAttrExists(*node, "value").ok() && in GetTensorFromConstNode()
680 TF_RETURN_IF_ERROR(CheckAttrExists(node, "T")); in MaterializeBroadcastGradientArgs()
852 TF_RETURN_IF_ERROR(CheckAttrExists(*input_node, "value")); in MaterializeConstantValuedNode()
1351 TF_RETURN_IF_ERROR(CheckAttrExists(*input_node, "value")); in EvaluateOneFoldable()
2474 !CheckAttrExists(*output_idx_node, "value").ok()) { in SimplifyCase()
2973 TF_RETURN_IF_ERROR(CheckAttrExists(*node, "T")); in SimplifyArithmeticOperations()
3127 if (!CheckAttrExists(parent, "T").ok()) return false; in PrepareConstantPushDown()
3335 if (!CheckAttrExists(*node, "T").ok()) return false; in ConstantPushDown()
|