Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for tensor_dtypes (0.06 sec)

  1. tensorflow/c/eager/gradients.cc

      std::vector<int64_t> tensor_ids(tensors.size());
      std::vector<tensorflow::DataType> tensor_dtypes(tensors.size());
      for (int i = 0; i < tensors.size(); i++) {
        tensor_ids[i] = ToId(tensors[i]);
        tensor_dtypes[i] = tensors[i]->DataType();
      }
      return GradientTape::ShouldRecord(tensor_ids, tensor_dtypes);
    }
    void Tape::DeleteTrace(const AbstractTensorHandle* t) {
      GradientTape::DeleteTrace(ToId(t));
    }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
Back to top