Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 418 for int64_t (0.39 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/eager/c_api_experimental_reader.cc

    template <typename... LabelType>
    int64_t TFE_MonitoringCounterReader::Read(const LabelType&... labels) {
      return counter->Read(labels...);
    }
    
    TFE_MonitoringCounterReader* TFE_MonitoringNewCounterReader(const char* name) {
      auto* result = new TFE_MonitoringCounterReader(name);
    
      return result;
    }
    
    int64_t TFE_MonitoringReadCounter0(TFE_MonitoringCounterReader* cell_reader) {
      int64_t result = cell_reader->Read();
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 20 03:14:47 GMT 2023
    - 1.4K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_experimental_reader_test.cc

    TFE_MonitoringCounter1* CreateCounter1(const char* counter_name,
                                           const char* label);
    void IncrementCounter0(TFE_MonitoringCounter0* counter, int64_t delta = 1);
    void IncrementCounter1(TFE_MonitoringCounter1* counter, const char* label,
                           int64_t delta = 1);
    
    TEST(CAPI, MonitoringCellReader0) {
      auto counter_name = "test/counter0";
      auto* counter = CreateCounter0(counter_name);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 20 03:14:47 GMT 2023
    - 2.9K bytes
    - Viewed (0)
  5. 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)
  6. tensorflow/c/eager/c_api_experimental_reader.h

    TFE_MonitoringNewCounterReader(const char* name);
    
    // Reads the value of a counter that was created with 0 labels.
    TF_CAPI_EXPORT extern int64_t TFE_MonitoringReadCounter0(
        TFE_MonitoringCounterReader*);
    
    // Reads the value of specific cell of a counter that was created with 1 label.
    TF_CAPI_EXPORT extern int64_t TFE_MonitoringReadCounter1(
        TFE_MonitoringCounterReader*, const char* label_value);
    
    #ifdef __cplusplus
    } /* end extern "C" */
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 20 03:14:47 GMT 2023
    - 2.3K bytes
    - Viewed (0)
  7. 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)
  8. tensorflow/c/eager/c_api_test_util.h

                                                      float data[], int64_t dims[],
                                                      int num_dims);
    
    // Get a Matrix TensorHandle with given float values and dimensions
    TFE_TensorHandle* TestTensorHandleWithDimsFloat(TFE_Context* ctx, float data[],
                                                    int64_t dims[], int num_dims);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Jul 17 23:43:59 GMT 2023
    - 7.7K bytes
    - Viewed (0)
  9. tensorflow/c/eager/parallel_device/parallel_device_lib.h

     private:
      ParallelTensor(const ParallelDevice& device,
                     std::vector<TensorHandlePtr> tensors,
                     absl::Span<const int64_t> shape, const TF_DataType dtype)
          : device_(device),
            tensors_(std::move(tensors)),
            shape_(std::vector<int64_t>(shape.begin(), shape.end())),
            dtype_(dtype) {}
      ParallelTensor(const ParallelDevice& device,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 25 15:21:13 GMT 2023
    - 12.9K bytes
    - Viewed (0)
  10. tensorflow/c/eager/tfe_monitoring_reader_internal.h

      explicit TFE_MonitoringCounterReader(const char* name) {
        counter = std::make_unique<
            ::tensorflow::monitoring::testing::CellReader<int64_t>>(name);
      }
      template <typename... LabelType>
      int64_t Read(const LabelType&... labels);
      std::unique_ptr<::tensorflow::monitoring::testing::CellReader<int64_t>>
          counter;
    };
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 20 03:14:47 GMT 2023
    - 1.3K bytes
    - Viewed (0)
Back to top