Home
last modified time | relevance | path

Searched refs:CheckAttrExists (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc363 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()
Dutils.cc477 Status CheckAttrExists(const NodeDef& node, const string& key) { in CheckAttrExists() function
487 TF_RETURN_IF_ERROR(CheckAttrExists(node, key)); in CheckAttrsExist()
Dutils.h269 Status CheckAttrExists(const NodeDef& node, const string& key);
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc1538 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()
Dconstant_folding.cc626 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()