- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for baek (0.16 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.h
std::vector<TFE_TensorHandle*> tensors() const { std::vector<TFE_TensorHandle*> result; result.reserve(tensors_.size()); for (const TensorHandlePtr& tensor : tensors_) { result.emplace_back(tensor.get()); } return result; } private: ParallelTensor(const ParallelDevice& device, std::vector<TensorHandlePtr> tensors,
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) -
tensorflow/c/eager/tape.h
// GradientTape which uses ones. Gradient* zero = input_tensors[target_index].ZerosLike(); new_zeros.push_back(zero); in_grads.push_back(zero); } else { in_grads.push_back(nullptr); } } else { in_grads.push_back(current_grad->second); } } // Avoid infinite recursion. Whichever forward function we run, it'll end up
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)