Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 16 for int64_t (0.18 sec)

  1. 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)
  2. 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)
  3. tensorflow/c/eager/c_api_unified_experimental.cc

      }
      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);
          return nullptr;
        }
      }
      tsl::Set_TF_Status_from_Status(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/gradients/grad_test_helper.cc

      }
    }
    
    void CheckTensorValue(AbstractTensorHandle* t, absl::Span<const float> manuals,
                          absl::Span<const int64_t> dims, double abs_error) {
      TF_Tensor* analytical_tensor;
      auto s = GetValue(t, &analytical_tensor);
      ASSERT_EQ(errors::OK, s.code()) << s.message();
    
      int64_t num_elem_analytical = 1;
      auto num_dims_analytical = TF_NumDims(analytical_tensor);
      ASSERT_EQ(dims.size(), num_dims_analytical);
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  5. tensorflow/c/eager/gradients.cc

      TapeVSpace vspace(ctx);
      std::vector<int64_t> target_tensor_ids = MakeTensorIDList(targets);
      std::vector<int64_t> source_tensor_ids = MakeTensorIDList(sources);
      tensorflow::gtl::FlatSet<int64_t> sources_set(source_tensor_ids.begin(),
                                                    source_tensor_ids.end());
      std::unordered_map<int64_t, TapeTensor> sources_that_are_targets;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  6. 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)
  7. tensorflow/c/eager/dlpack.cc

    }
    
    // Checks whether the stride array matches the layout of compact, row-majored
    // data.
    bool IsValidStrideCompactRowMajorData(int64_t* shape_arr, int64_t* stride_arr,
                                          int ndim) {
      bool valid = true;
      int64_t expected_stride = 1;
      for (int i = ndim - 1; i >= 0; --i) {
        // Empty tensors are always compact regardless of strides.
        if (shape_arr[i] == 0) return true;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  8. tensorflow/c/experimental/gradients/nn_grad_test.cc

      ASSERT_EQ(errors::OK, status_.code()) << status_.message();
    
      auto ReluGradModel = BuildGradModel(ReluModel, registry_);
    
      float X_vals[] = {1.0f, 2.0f, 3.0f, -5.0f, -4.0f, -3.0f, 2.0f, 10.0f, -1.0f};
      int64_t X_dims[] = {3, 3};
      AbstractTensorHandlePtr X;
      {
        AbstractTensorHandle* X_raw;
        status_ = TestTensorHandleWithDims<float, TF_FLOAT>(
            immediate_execution_ctx_.get(), X_vals, X_dims, 2, &X_raw);
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 8.3K bytes
    - Viewed (0)
  9. 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)
  10. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

                        const TFE_OpAttrs* attributes, int expected_max_outputs,
                        CancellationManager& cancellation_manager,
                        absl::optional<int64_t> step_id = absl::nullopt);
      // Block until the previous `StartExecute` operation has executed. Forwards
      // the status from `TFE_Execute` and returns outputs if the status is OK.
    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)
Back to top