Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for Int64 (0.31 sec)

  1. tensorflow/c/c_api_test.cc

      }
    }
    
    void TestEncodeDecode(int line, const std::vector<string>& data) {
      const int64_t n = data.size();
      Status status;
      for (const std::vector<int64_t>& dims :
           std::vector<std::vector<int64_t>>{{n}, {1, n}, {n, 1}, {n / 2, 2}}) {
        // Create C++ Tensor
        Tensor src(tensorflow::DT_STRING, TensorShape(dims));
        for (int64_t i = 0; i < src.NumElements(); ++i) {
          src.flat<tstring>()(i) = data[i];
        }
    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)
  2. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        return absl::OkStatus();
      }
      Status SetAttrShape(const char* attr_name, const int64_t* dims,
                          const int num_dims) override {
        PartialTensorShape shape;
        if (num_dims >= 0) {
          shape = PartialTensorShape(ArraySlice<int64_t>(
              reinterpret_cast<const int64_t*>(dims), num_dims));
        }
        op_->node_builder.Attr(attr_name, shape);
        return absl::OkStatus();
      }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  3. tensorflow/c/c_api.cc

                       int64_t value) {
      desc->node_builder.Attr(attr_name, static_cast<int64_t>(value));
    }
    
    void TF_SetAttrIntList(TF_OperationDescription* desc, const char* attr_name,
                           const int64_t* values, int num_values) {
      desc->node_builder.Attr(
          attr_name, ArraySlice<const int64_t>(
                         reinterpret_cast<const int64_t*>(values), num_values));
    }
    
    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)
  4. tensorflow/c/eager/tape.h

      // Map from tensor ID to how many references still exist for this tensor in
      // the tape.
      std::unordered_map<int64_t, int64_t> tensor_usage_counts;
    
      // Maps from op ID to how many output tensors of this op still need to have
      // their gradients computed.
      std::unordered_map<int64_t, int64_t> op_missing_tensor;
    };
    
    // If `persistent_tape` is true, op_tape is not changed and none of the
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  5. tensorflow/c/eager/c_api.cc

      return num_dims;
    }
    
    int64_t TFE_TensorHandleNumElements(TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return -1;
      }
    
      int64_t num_elements = -1;
      status->status = tensorflow::unwrap(h)->NumElements(&num_elements);
      return num_elements;
    }
    
    int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  6. tensorflow/c/eager/c_api_experimental.cc

    }
    
    TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx, TF_DataType dtype,
                                      const int64_t* dims, int num_dims,
                                      TF_Status* status) {
      std::vector<int64_t> dimvec(num_dims);
      for (int i = 0; i < num_dims; ++i) {
        dimvec[i] = static_cast<int64_t>(dims[i]);
      }
    
      if (ctx == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid Context");
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
  7. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc

          requested_read_length = INT32_MAX;
        else
          requested_read_length = n;
    
        // `pread` returns a `ssize_t` on POSIX, but due to interface being
        // cross-platform, return type of `Read` is `int64_t`.
        int64_t r = int64_t{pread(posix_file->fd, dst, requested_read_length,
                                  static_cast<off_t>(offset))};
        if (r > 0) {
          dst += r;
          offset += static_cast<uint64_t>(r);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Mar 24 20:08:23 GMT 2024
    - 15.8K bytes
    - Viewed (0)
  8. tensorflow/c/c_api_experimental.cc

                                     const int64_t* dims, int num_dims) {
      DCHECK(index >= 0 && index < shape_list->num_items);
      TF_ShapeAndType& shape = shape_list->items[index];
      DCHECK(shape.dims == nullptr) << "Shape at " << index << " is already set!";
      DCHECK(num_dims >= 0) << "Number of dimensions cannot be negative!";
      shape.num_dims = num_dims;
      shape.dims = new int64_t[num_dims];
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  9. RELEASE.md

            *   `tf.where` op for data types
                `tf.int32`/`tf.uint32`/`tf.int8`/`tf.uint8`/`tf.int64`.
            *   `tf.random.normal` op for output data type `tf.float32` on CPU.
            *   `tf.random.uniform` op for output data type `tf.float32` on CPU.
            *   `tf.random.categorical` op for output data type `tf.int64` on CPU.
    
    *   `tensorflow.experimental.tensorrt`:
    
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Mon Apr 29 19:17:57 GMT 2024
    - 727.7K bytes
    - Viewed (8)
Back to top