Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 32 for shape_1 (0.1 sec)

  1. tensorflow/c/c_api_internal.h

    namespace tensorflow {
    
    // 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.
    //
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat May 13 00:49:12 UTC 2023
    - 7.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/immediate_execution_tensor_handle.h

      //
      // -1 indicates an unknown axis length; this is unreachable for most standard
      // ImmediateExecutionTensorHandles, but comes up for example when computing
      // the shape of a parallel tensor with component shapes differing across
      // devices.
      virtual absl::Status Dim(int dim_index, int64_t* dim) const = 0;
    
      // Returns the device which created the handle.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test_util.h

    TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
    
    // Return an identity op.
    TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a);
    
    // Return a shape op fetching the shape of `a`.
    TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a);
    
    // Return an allreduce op adding up input tensor `in` from `group_size` workers.
    TFE_Op* AllReduceOp(TFE_Context* ctx, TFE_TensorHandle* in, int group_size);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Jul 17 23:43:59 UTC 2023
    - 7.7K bytes
    - Viewed (0)
  4. tensorflow/c/checkpoint_reader.h

     public:
      CheckpointReader(const string& filename, TF_Status* status);
    
      bool HasTensor(const string& name) const;
      const string DebugString() const;
    
      // Returns a map from variable names to their shapes.  Slices of a partitioned
      // tensor are combined into a single entry.
      const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const;
    
      // Returns a map from variable names to their data types.  Slices of a
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 12 08:49:52 UTC 2023
    - 3.1K bytes
    - Viewed (0)
  5. tensorflow/c/eager/dlpack_test.cc

      dltensor_in->device = {kDLCPU, 0};
      dltensor_in->ndim = static_cast<int32_t>(shape.size());
      dltensor_in->dtype = {kDLFloat, 32, 1};
      dltensor_in->shape = shape.data();
      dltensor_in->strides = strides.data();
      TFE_TensorHandle* handle = TFE_HandleFromDLPack(&dlm_in, status, ctx);
      ASSERT_NE(handle, nullptr)
          << TF_Message(status) << " (shape=[" << absl::StrJoin(shape, ",")
          << "], strides=[" << absl::StrJoin(strides, ",") << "])";
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Jun 30 03:04:46 UTC 2023
    - 4.4K bytes
    - Viewed (0)
  6. tensorflow/c/eager/abstract_tensor_handle.h

      virtual absl::Status TensorHandleStatus() const;
    
      // Returns tensor shape. If tensor has unknown rank, shape remains untouched.
      virtual absl::Status Shape(tensorflow::PartialTensorShape* shape) const = 0;
    
      // Returns tensor (full) type.
      // While there is no immediate plan to deprecate dtype and shape in favor
      // of only using full type type information, this is a future possibility.
      //
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_debug.cc

      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);
      }
      return shape;
    }
    
    }  // namespace
    
    extern "C" {
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 2.5K bytes
    - Viewed (0)
  8. tensorflow/c/eager/abstract_tensor_handle.cc

    namespace tensorflow {
    
    std::string AbstractTensorHandle::DebugString() const {
      PartialTensorShape shape;
      absl::Status s = Shape(&shape);
      std::string shape_string;
      if (!s.ok()) {
        shape_string = "<error computing shape>";
      } else {
        shape_string = shape.DebugString();
      }
      return absl::StrCat("TensorHandle(shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()),
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 1.4K bytes
    - Viewed (0)
  9. tensorflow/c/eager/immediate_execution_tensor_handle.cc

    namespace tensorflow {
    
    std::string ImmediateExecutionTensorHandle::DebugString() const {
      PartialTensorShape shape;
      std::string shape_string;
      if (Shape(&shape).ok()) {
        shape_string = shape.DebugString();
      } else {
        shape_string = "<error computing shape>";
      }
      std::string value_string;
      if (!SummarizeValue(value_string).ok()) {
        value_string = "<error computing value>";
      }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:24:07 UTC 2024
    - 2.1K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_unified_experimental.cc

        return nullptr;
      }
      tensorflow::PartialTensorShape partial_shape;
      if (shape.num_dims != -1) {
        DCHECK(shape.dim_sizes != nullptr);
        Status status = tensorflow::PartialTensorShape::MakePartialShape(
            reinterpret_cast<int64_t*>(shape.dim_sizes), shape.num_dims,
            &partial_shape);
        if (!status.ok()) {
          tsl::Set_TF_Status_from_Status(s, status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
Back to top