Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 269 for int64_t (0.17 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. tensorflow/c/eager/c_api_test_util.cc

      return th;
    }
    
    TFE_TensorHandle* DoubleTestMatrixTensorHandle(TFE_Context* ctx) {
      int64_t dims[] = {2, 2};
      double data[] = {1.0, 2.0, 3.0, 4.0};
      TF_Status* status = TF_NewStatus();
      TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_DOUBLE, &dims[0],
                                            sizeof(dims) / sizeof(int64_t), status);
      memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  10. 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)
Back to top