Searched refs:requested_shapes (Results 1 – 3 of 3) sorted by relevance
418 std::vector<TensorShape> requested_shapes; in MatchesPriorityNodeDefShapes() local419 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, "shapes", &requested_shapes)); in MatchesPriorityNodeDefShapes()420 requested_shapes.insert(requested_shapes.begin(), TensorShape({})); in MatchesPriorityNodeDefShapes()421 if (requested_shapes != component_shapes_) { in MatchesPriorityNodeDefShapes()426 ShapeListString(requested_shapes)); in MatchesPriorityNodeDefShapes()
130 std::vector<TensorShape> requested_shapes; in MatchesNodeDefShapes() local131 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, "shapes", &requested_shapes)); in MatchesNodeDefShapes()132 if (requested_shapes != component_shapes_) { in MatchesNodeDefShapes()137 ShapeListString(requested_shapes)); in MatchesNodeDefShapes()
270 std::vector<PartialTensorShape> requested_shapes; in CompatibleNodeDefShapes() local271 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, "shapes", &requested_shapes)); in CompatibleNodeDefShapes()272 if (!PartialTensorShapeUtils::AreCompatible(requested_shapes, in CompatibleNodeDefShapes()278 PartialTensorShapeUtils::PartialShapeListString(requested_shapes)); in CompatibleNodeDefShapes()