Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 20 for sank (0.31 sec)

  1. tensorflow/c/eager/c_api_debug.cc

                                             tensorflow::Status* status) {
      std::vector<int64_t> shape;
      int rank = -1;
      *status = handle.NumDims(&rank);
      if (!status->ok()) {
        return shape;
      }
      shape.reserve(rank);
      for (int i = 0; i < rank; ++i) {
        int64_t dim;
        *status = handle.Dim(i, &dim);
        if (!status->ok()) {
          return shape;
        }
        shape.push_back(dim);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Aug 11 01:20:50 GMT 2021
    - 2.5K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/gradients/tape/tape_operation.cc

                                       " dimensions which is over the limit of ",
                                       TensorShape::MaxDimensions(), ".");
      }
      TensorShapeProto proto;
      if (num_dims < 0) {
        proto.set_unknown_rank(true);
      } else {
        for (int d = 0; d < num_dims; ++d) {
          proto.add_dim()->set_size(dims[d]);
        }
      }
    
      forward_op_.attrs.Set(attr_name, proto);
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  3. tensorflow/c/c_api_internal.h

    // Set the shapes and types of the output's handle.
    //
    // The lengths of the arrays pointed to by `shapes`, `ranks`, and `types` must
    // all be equal to `num_shapes_and_types`. If `ranks[i] != -1`, (i.e., if the
    // rank is known), then it must be equal to the length of `shapes[i]`; if
    // `ranks[i] == 1`, then `shapes[i]` may be nullptr.
    //
    // TODO(akshayka): Implement a corresponding getter method.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sat May 13 00:49:12 GMT 2023
    - 7.6K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
      EXPECT_EQ(dims[0], returned_dims[0]);
      EXPECT_EQ(dims[1], returned_dims[1]);
    
      // Try to set 'unknown' with unknown rank on the shape and see that
      // it doesn't change.
      TF_GraphSetTensorShape(graph, feed_out_0, /*dims=*/nullptr, -1, s);
      EXPECT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  5. tensorflow/c/c_api.cc

      }
    
      tensorflow::shape_inference::ShapeHandle shape = ic->output(output.index);
    
      int rank = -1;
      if (ic->RankKnown(shape)) {
        rank = ic->Rank(shape);
      }
    
      if (num_dims != rank) {
        status->status = InvalidArgument("Expected rank is ", num_dims,
                                         " but actual rank is ", rank);
        return;
      }
    
      if (num_dims == 0) {
        // Output shape is a scalar.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/grappler/grappler_test.cc

          Status s = tensorflow::BufferToMessage(in_props_buf[0], &in_props);
          TF_ASSERT_OK(s);
    
          EXPECT_EQ(DT_FLOAT, in_props.dtype());
          EXPECT_FALSE(in_props.shape().unknown_rank());
          EXPECT_EQ(2, in_props.shape().dim_size());
          EXPECT_EQ(10, in_props.shape().dim(0).size());
          EXPECT_EQ(1, in_props.shape().dim(1).size());
    
          for (int i = 0; i < in_props_buf.size(); i++)
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Thu Apr 13 22:30:58 GMT 2023
    - 11.6K bytes
    - Viewed (0)
  7. tensorflow/c/eager/parallel_device/parallel_device_testlib.h

      // dtype of read operations).
      TF_DataType type_;
    };
    
    // Creates a TFE_TensorHandle with value `v`.
    TensorHandlePtr FloatTensorHandle(float v, TF_Status* status);
    
    // Creates a rank-one TFE_TensorHandle with value `v`.
    TensorHandlePtr VectorFloatTensorHandle(const std::vector<float>& v,
                                            TF_Status* status);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 09 01:12:35 GMT 2021
    - 6.9K bytes
    - Viewed (0)
  8. tensorflow/c/eager/parallel_device/parallel_device_testlib.cc

          TF_NewTensor(TF_FLOAT, nullptr, 0, values, num_bytes, &FloatDeallocator,
                       nullptr),
          TF_DeleteTensor);
      return TensorHandlePtr(TFE_NewTensorHandle(tensor.get(), status));
    }
    
    // Creates a rank-one TFE_TensorHandle with value `v`.
    TensorHandlePtr VectorFloatTensorHandle(const std::vector<float>& v,
                                            TF_Status* status) {
      const int num_bytes = v.size() * sizeof(float);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Jun 15 15:44:44 GMT 2021
    - 12.5K bytes
    - Viewed (0)
  9. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

            return errors::Unimplemented(absl::StrCat(
                "Computing the shape of a ParallelTensor when the components do "
                "not all have the same rank is not supported. One tensor had "
                "shape ",
                first_shape.DebugString(), " and another had shape ",
                component_shape.DebugString()));
          } else {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  10. tensorflow/c/c_api.h

                                                  const char* attr_name,
                                                  const char* value, size_t length);
    
    // Set `num_dims` to -1 to represent "unknown rank".  Otherwise,
    // `dims` points to an array of length `num_dims`.  `dims[i]` must be
    // >= -1, with -1 meaning "unknown dimension".
    TF_CAPI_EXPORT extern void TF_SetAttrShape(TF_OperationDescription* desc,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
Back to top