- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for rack (0.31 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) -
tensorflow/c/c_api_experimental.h
TFE_Context* ctx, const char* task, int64_t timeout_in_ms, TF_Status* status); // Information about the shape of a Tensor and its type. struct TF_ShapeAndType { // Number of dimensions. -1 indicates unknown rank. int num_dims; // Array of dimensions. -1 indicates unknown dim. int64_t* dims; // The data type. May be 0 to denote unknown type. TF_DataType dtype; };
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0)