Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 40 for int64_t (0.2 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/eager/dlpack_test.cc

    #include "tensorflow/core/platform/test.h"
    
    namespace tensorflow {
    namespace {
    
    void TestHandleFromDLPack(TF_Status* status, TFE_Context* ctx,
                              std::vector<int64_t> shape,
                              std::vector<int64_t> strides) {
      size_t num_elements = 1;
      for (int i = 0; i < static_cast<int32_t>(shape.size()); ++i) {
        num_elements *= shape[i];
      }
      std::vector<float> data(num_elements);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Jun 30 03:04:46 GMT 2023
    - 4.4K bytes
    - Viewed (0)
  3. 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)
  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/experimental/gradients/math_grad_test.cc

      // well with `MatMul` and remove `TestMatMul*` in
      // `mnist_gradients_test` when done.
      GTEST_SKIP();
    
      float A_vals[] = {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f};
      int64_t A_dims[] = {3, 3};
      AbstractTensorHandlePtr A;
      {
        AbstractTensorHandle* A_raw;
        status_ = TestTensorHandleWithDims<float, TF_FLOAT>(
            immediate_execution_ctx_.get(), A_vals, A_dims, 2, &A_raw);
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Thu Apr 13 17:32:14 GMT 2023
    - 16.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/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)
  8. tensorflow/c/eager/c_api_debug.cc

    namespace {
    
    std::vector<int64_t> TensorShapeAsVector(const tensorflow::TensorHandle& handle,
                                             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);
    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)
  9. tensorflow/c/experimental/gradients/tape/tape_operation.cc

    }
    Status TapeOperation::SetAttrIntList(const char* attr_name,
                                         const int64_t* values, int num_values) {
      forward_op_.attrs.Set(
          attr_name, gtl::ArraySlice<const int64_t>(
                         reinterpret_cast<const int64_t*>(values), num_values));
      return parent_op_->SetAttrIntList(attr_name, values, num_values);
    }
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  10. 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)
Back to top