- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for Mack (0.14 sec)
-
tensorflow/c/eager/c_api_experimental.h
// custom device. // // Many devices will want to simply return an "unimplemented" status // here. This is the default behavior if `pack` is null when passed to // TFE_RegisterCustomDevice. TFE_TensorHandle* (*pack)(TFE_Context* context, TFE_TensorHandle** handles, int num_handles, TF_Status* s, void* device_info) = nullptr;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0) -
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)