HomeSort by relevance Sort by last modified time
    Searched refs:requested_shapes (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
priority_queue.cc 418 std::vector<TensorShape> requested_shapes; local
419 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, "shapes", &requested_shapes));
420 requested_shapes.insert(requested_shapes.begin(), TensorShape({}));
421 if (requested_shapes != component_shapes_) {
426 ShapeListString(requested_shapes));
queue_base.cc 130 std::vector<TensorShape> requested_shapes; local
131 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, "shapes", &requested_shapes));
132 if (requested_shapes != component_shapes_) {
137 ShapeListString(requested_shapes));
padding_fifo_queue.cc 270 std::vector<PartialTensorShape> requested_shapes; local
271 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, "shapes", &requested_shapes));
272 if (!PartialTensorShapeUtils::AreCompatible(requested_shapes,
278 PartialTensorShapeUtils::PartialShapeListString(requested_shapes));

Completed in 479 milliseconds