- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for VarHandleOp (0.2 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
"/job:localhost/replica:0/task:0/device:CPU:1"}; ParallelDevice parallel_device(std::move(devices)); std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op( TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_test.cc
// Create a variable handle placed on the custom device. 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());
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
const int64_t* dims, const int num_dims, const char* device, TF_Status* status) { std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( TFE_NewOp(context, "VarHandleOp", status), TFE_DeleteOp); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrType(op.get(), "dtype", type); TFE_OpSetAttrShape(op.get(), "shape", dims, num_dims, status);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
TFE_DeleteContextOptions(opts); TFE_Op* var_op = TFE_NewOp(ctx, "VarHandleOp", status); TFE_OpSetAttrType(var_op, "dtype", TF_INT64); TFE_OpSetAttrShape(var_op, "shape", {}, 0, status); const TFE_OpAttrs* attributes = TFE_OpGetAttrs(var_op); TFE_Op* copy_op = TFE_NewOp(ctx, "VarHandleOp", status); TFE_OpSetAttrType(copy_op, "dtype", TF_FLOAT); TFE_OpAddAttrs(copy_op, attributes);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/eager/c_api_test_util.cc
TFE_TensorHandle* TestVariable(TFE_Context* ctx, float value, const tensorflow::string& device_name) { TF_Status* status = TF_NewStatus(); // Create the variable handle. TFE_Op* op = TFE_NewOp(ctx, "VarHandleOp", status); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrType(op, "dtype", TF_FLOAT); TFE_OpSetAttrShape(op, "shape", {}, 0, status); TFE_OpSetAttrString(op, "container", "localhost", 0);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2)