- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 80 for Op (0.13 sec)
-
tensorflow/c/eager/c_api_experimental_test.cc
int num_retvals = 1; TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx_1, "ReadVariableOp", status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpSetAttrType(op, "dtype", TF_FLOAT); TFE_OpAddInput(op, var_handle, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_Execute(op, &handle_1, &num_retvals, status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/eager/c_api_remote_test_util.h
#define TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_ // Run a function containing a MatMul op and check its output. // If heavy_load_on_streaming_rpc is true, send some rpc requests before the one // which creates a remote input, to simulate a scenario that the remote input // is not ready when we start running an op or a function. void TestRemoteExecuteSilentCopies(bool async, bool remote, bool func,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Dec 11 22:56:03 GMT 2020 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api.h
// the additional sanity checks there seem unnecessary; typedef struct TFE_Op TFE_Op; TF_CAPI_EXPORT extern TFE_Op* TFE_NewOp(TFE_Context* ctx, const char* op_or_function_name, TF_Status* status); TF_CAPI_EXPORT extern void TFE_DeleteOp(TFE_Op* op); // Returns the op or function name `op` will execute. //
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 22.8K bytes - Viewed (1) -
tensorflow/c/eager/gradients.cc
namespace internal { Status Reset(AbstractOperation* op_, const char* op, const char* raw_device_name, ForwardOperation* forward_op_) { forward_op_->op_name = op; forward_op_->attrs.Reset(op); return op_->Reset(op, raw_device_name); } Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input, ForwardOperation* forward_op_) { TF_RETURN_IF_ERROR(op_->AddInput(input));
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) -
tensorflow/c/eager/c_api_unified_experimental.cc
void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type, TF_Status* s) { tsl::Set_TF_Status_from_Status( s, unwrap(op)->Reset(op_type, /*raw_device_name=*/nullptr)); } void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name, TF_Status* s) { TracingOperation* tracing_op = dyn_cast<TracingOperation>(unwrap(op));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
TF_Graph* graph, TF_Status* s, const char* name, TF_Operation** op, const string& op_device, bool check) { TF_OperationDescription* desc = TF_NewOperation(graph, op_name, name); if (!op_device.empty()) { TF_SetDevice(desc, op_device.c_str()); } TF_AddInput(desc, {l, 0}); TF_AddInput(desc, {r, 0}); *op = TF_FinishOperation(desc, s);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/eager/custom_device_test.cc
std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); TFE_OpSetAttrShape(op.get(), "shape", {}, 0, status.get()); TFE_OpSetAttrString(op.get(), "container", "", 0);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.h
absl::Span<AbstractTensorHandle* const>, absl::Span<AbstractTensorHandle*>)>; // Runs `model` maybe wrapped in a function call op. This can be thought as // being equivalent to the following python code. // // if use_function: // outputs = tf.function(model)(inputs) // else: // outputs = model(inputs)
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.h
// TODO(srbs): Add APIs for specifying attrs etc. // `op_type` must outlive `op`. void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type, TF_Status* s); // `op_name` must outlive `op`. void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name, TF_Status* s); // `attr_name` must outlive `op`.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:00 GMT 2021 - 7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
GradientFunction* ReluRegisterer(const ForwardOperation& op) { return new ReluGradientFunction(op.outputs); } GradientFunction* SparseSoftmaxCrossEntropyWithLogitsRegisterer( const ForwardOperation& op) { return new SparseSoftmaxCrossEntropyWithLogitsGradientFunction(op.outputs); } GradientFunction* BiasAddRegisterer(const ForwardOperation& op) { return new BiasAddGradientFunction(op.attrs); } } // namespace gradients
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0)