Searched refs:CheckAttrExists (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/ |
D | utils_test.cc | 363 TEST(CheckAttrExists, All) { in TEST() argument 369 TF_EXPECT_OK(CheckAttrExists(node, "apple")); in TEST() 370 TF_EXPECT_OK(CheckAttrExists(node, "pear")); in TEST() 378 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 | 269 Status CheckAttrExists(const NodeDef& node, const string& key);
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | arithmetic_optimizer.cc | 1538 TF_RETURN_IF_ERROR(CheckAttrExists(node, "N")); in InitializeChains() 2119 TF_RETURN_IF_ERROR(CheckAttrExists(*weights, "dtype")); in TrySimplify() 2917 TF_RETURN_IF_ERROR(CheckAttrExists(*root, "T")); in TrySimplify() 3110 TF_RETURN_IF_ERROR(CheckAttrExists(*pack, "axis")); in CheckInputs() 3169 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_begin, "value")); in GetSliceAxis() 3173 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_end, "value")); in GetSliceAxis() 3177 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_strides, "value")); in GetSliceAxis()
|
D | constant_folding.cc | 626 TF_RETURN_IF_ERROR(CheckAttrExists(node, "T")); in MaterializeBroadcastGradientArgs() 794 TF_RETURN_IF_ERROR(CheckAttrExists(*input_node, "value")); in MaterializeConstantValuedNode() 1191 TF_RETURN_IF_ERROR(CheckAttrExists(*input_node, "value")); in EvaluateOneFoldable() 2686 TF_RETURN_IF_ERROR(CheckAttrExists(*node, "T")); in SimplifyArithmeticOperations()
|