Home
last modified time | relevance | path

Searched refs:requested_shapes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpriority_queue.cc418 std::vector<TensorShape> requested_shapes; in MatchesPriorityNodeDefShapes() local
419 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()
Dqueue_base.cc130 std::vector<TensorShape> requested_shapes; in MatchesNodeDefShapes() local
131 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()
Dpadding_fifo_queue.cc270 std::vector<PartialTensorShape> requested_shapes; in CompatibleNodeDefShapes() local
271 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()