- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for getContext (0.21 sec)
-
tensorflow/c/c_api.cc
TF_Status* status) { Node* node = &output.oper->node; mutex_lock l(graph->mu); tensorflow::shape_inference::InferenceContext* ic = graph->refiner.GetContext(node); if (ic == nullptr) { status->status = InvalidArgument("Node ", node->name(), " was not found in the graph"); return; } auto shape_and_type_vec =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_operation.h
const = 0; virtual Status SetInput(size_t index, ImmediateExecutionTensorHandle* input) = 0; virtual ImmediateExecutionContext* GetContext() const = 0; // Following two methods are used to support custom device. // Return true if the inputs contain custom device tensor handle. It means // that the argument need to be handled by a custom device.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 22:40:32 GMT 2022 - 3.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
return tensorflow::unwrap(op)->Name().c_str(); } TFE_Context* TFE_OpGetContext(const TFE_Op* op, TF_Status* status) { return tensorflow::wrap(tensorflow::unwrap(op)->GetContext()); } void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) { status->status = tensorflow::unwrap(op)->SetDeviceName(device_name); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)