- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for input_dtypes (5.82 sec)
-
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) -
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)