Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 55 for B_dims (0.05 sec)

  1. tensorflow/c/eager/abstract_operation.h

      virtual absl::Status SetAttrBool(const char* attr_name, bool value) = 0;
      virtual absl::Status SetAttrType(const char* attr_name, DataType value) = 0;
      virtual absl::Status SetAttrShape(const char* attr_name, const int64_t* dims,
                                        const int num_dims) = 0;
      virtual absl::Status SetAttrShape(const char* attr_name,
                                        const PartialTensorShape shape);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.cc

      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);
      return op_->SetAttrShape(attr_name, dims, num_dims);
    }
    absl::Status SetAttrFunction(AbstractOperation* op_, const char* attr_name,
                                 const AbstractOperation* value,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.cc

          } else {
            const auto num_dims = tensor_shape.dim_size();
            std::unique_ptr<int64_t[]> dims(new int64_t[num_dims]);
            for (int i = 0; i < num_dims; ++i) {
              dims[i] = tensor_shape.dim(i).size();
            }
            TFE_OpSetAttrShape(op, attr_name, dims.get(), num_dims, status);
          }
        } break;
        case tensorflow::AttrValue::kFunc: {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  4. tensorflow/c/c_api.h

    // Returns the shape of the Tensor referenced by `output` in `graph`
    // into `dims`. `dims` must be an array large enough to hold `num_dims`
    // entries (e.g., the return value of TF_GraphGetTensorNumDims).
    //
    // If the number of dimensions in the shape is unknown or the shape is
    // a scalar, `dims` will remain untouched. Otherwise, each element of
    // `dims` will be set corresponding to the size of the dimension. An
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental.h

    TF_CAPI_EXPORT extern TF_DataType TF_CheckpointReaderGetVariableDataType(
        TF_CheckpointReader* reader, const char* name);
    // Read the shape of a variable and write to `dims`
    TF_CAPI_EXPORT extern void TF_CheckpointReaderGetVariableShape(
        TF_CheckpointReader* reader, const char* name, int64_t* dims, int num_dims,
        TF_Status* status);
    // Get the number of dimension of a variable
    TF_CAPI_EXPORT extern int TF_CheckpointReaderGetVariableNumDims(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
  6. tensorflow/c/eager/gradients_internal.h

    absl::Status SetAttrType(AbstractOperation*, const char* attr_name,
                             DataType value, ForwardOperation*);
    absl::Status SetAttrShape(AbstractOperation*, const char* attr_name,
                              const int64_t* dims, const int num_dims,
                              ForwardOperation*);
    absl::Status SetAttrFunction(AbstractOperation*, const char* attr_name,
                                 const AbstractOperation* value, ForwardOperation*);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  7. tensorflow/c/eager/dlpack.cc

      DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor;
      dlm_tensor->manager_ctx = tf_dlm_tensor_ctx;
      dlm_tensor->deleter = &DLManagedTensorDeleter;
      dlm_tensor->dl_tensor.device = tf_dlm_context;
      int ndim = tensor->dims();
      dlm_tensor->dl_tensor.ndim = ndim;
      dlm_tensor->dl_tensor.data = tf_dlm_data;
      dlm_tensor->dl_tensor.dtype = tf_dlm_type;
    
      std::vector<int64_t>* shape_arr = &tf_dlm_tensor_ctx->shape;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.9K bytes
    - Viewed (0)
  8. tensorflow/c/c_api.cc

      static char empty;
      int64_t nelems = 1;
      std::vector<int64_t> dims;
      dims.reserve(shape.dims());
      for (int i = 0; i < shape.dims(); ++i) {
        dims.push_back(shape.dim_size(i));
        nelems *= shape.dim_size(i);
      }
      CHECK_EQ(nelems, 0);
      return TF_NewTensor(
          dtype, reinterpret_cast<const int64_t*>(dims.data()), shape.dims(),
          reinterpret_cast<void*>(&empty), 0, [](void*, size_t, void*) {}, nullptr);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api.h

    // If the number of dimensions is unknown, `num_dims` must be set to
    // -1 and `dims` can be null.  If a dimension is unknown, the
    // corresponding entry in the `dims` array must be -1.
    TF_CAPI_EXPORT extern void TFE_OpSetAttrShape(TFE_Op* op, const char* attr_name,
                                                  const int64_t* dims,
                                                  const int num_dims,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  10. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

          PartialTensorShape component_shape;
          TF_RETURN_IF_ERROR(unwrap(component.get())->Shape(&component_shape));
          if (combined_shape.dims() < 0 ||
              combined_shape.dims() != component_shape.dims()) {
            PartialTensorShape first_shape;
            TF_RETURN_IF_ERROR(unwrap(tensors_[0].get())->Shape(&first_shape));
            return errors::Unimplemented(absl::StrCat(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
Back to top