Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TensorShape (0.63 sec)

  1. tensorflow/c/experimental/gradients/tape/tape_operation.cc

                                       const int num_dims) {
      if (num_dims > TensorShape::MaxDimensions()) {
        return errors::InvalidArgument("Value specified for `", attr_name, "` has ",
                                       num_dims,
                                       " dimensions which is over the limit of ",
                                       TensorShape::MaxDimensions(), ".");
      }
      TensorShapeProto proto;
      if (num_dims < 0) {
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  2. tensorflow/c/checkpoint_reader.cc

        TF_Status* out_status) const {
      Status status;
      if (reader_ != nullptr) {
        status = reader_->GetTensor(name, out_tensor);
      } else {
        tensorflow::DataType dtype;
        tensorflow::TensorShape shape;
        status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape);
        if (status.ok()) {
          out_tensor->reset(new Tensor(dtype, shape));
          status = v2_reader_->Lookup(name, out_tensor->get());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Aug 25 21:29:12 GMT 2023
    - 5.5K bytes
    - Viewed (0)
Back to top