Searched refs:DimKnownRank (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 967 if (!InferenceContext::DimKnownRank(s1, i).SameHandle( in EquivalentShapes()
968 InferenceContext::DimKnownRank(s2, i))) { in EquivalentShapes()
970 InferenceContext::Value(InferenceContext::DimKnownRank(s1, i)); in EquivalentShapes()
972 InferenceContext::Value(InferenceContext::DimKnownRank(s2, i)); in EquivalentShapes()
1002 if (!InferenceContext::DimKnownRank(inferred_shape, i) in CompatibleShapes()
1004 InferenceContext::DimKnownRank(annotated_shape, i))) { in CompatibleShapes()
1006 InferenceContext::DimKnownRank(inferred_shape, i)); in CompatibleShapes()
1008 InferenceContext::DimKnownRank(annotated_shape, i)); in CompatibleShapes()
1724 TF_RETURN_IF_ERROR(dims_.Merge(InferenceContext::DimKnownRank(s1, i), in Merge()
1725 InferenceContext::DimKnownRank(s2, i))); in Merge()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 346 return DimKnownRank(s, idx); in Dim()
349 static DimensionHandle DimKnownRank(ShapeHandle s, int64 idx) { in DimKnownRank() function
|