Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for input_dtypes (0.17 sec)

  1. tensorflow/c/eager/tape.h

        absl::Span<const int64_t> input_tensor_id,
        absl::Span<const tensorflow::DataType> input_dtypes,
        const std::function<BackwardFunction*()>& backward_function_getter,
        const std::function<void(BackwardFunction*)>& backward_function_deleter) {
      if (!ShouldRecord(input_tensor_id, input_dtypes)) {
        return;
      }
      std::vector<int64_t> ids;
      ids.reserve(input_tensor_id.size());
    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

                               const string& op_name) {
      std::vector<int64_t> input_ids(inputs.size());
      std::vector<tensorflow::DataType> input_dtypes(inputs.size());
      for (int i = 0; i < inputs.size(); i++) {
        input_ids[i] = ToId(inputs[i]);
        input_dtypes[i] = inputs[i]->DataType();
      }
      std::vector<TapeTensor> tape_tensors;
      tape_tensors.reserve(outputs.size());
      for (auto t : outputs) {
    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