- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for AssignVariableOp (0.16 sec)
-
tensorflow/c/eager/custom_device_test.cc
std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> one( TestScalarTensorHandle(context.get(), 111.f), TFE_DeleteTensorHandle); op.reset(TFE_NewOp(context.get(), "AssignVariableOp", status.get())); TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); TFE_OpAddInput(op.get(), var_handle, status.get()); TFE_OpAddInput(op.get(), one.get(), status.get());
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/parallel_device/parallel_device_testlib.cc
GeneralAssignment("AssignAddVariableOp", context, value, status); } void Variable::Assign(TFE_Context* context, TFE_TensorHandle* value, TF_Status* status) { GeneralAssignment("AssignVariableOp", context, value, status); } // Passed to `TF_NewTensor` to indicate how an array of floats should be // deleted. static void FloatDeallocator(void* data, size_t, void* arg) {
C++ - Registered: Tue Apr 30 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_util.cc
if (TF_GetCode(status) != TF_OK) return nullptr; TFE_DeleteOp(op); if (TF_GetCode(status) != TF_OK) return nullptr; CHECK_EQ(1, num_retvals); // Assign 'value' to it. op = TFE_NewOp(ctx, "AssignVariableOp", status); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrType(op, "dtype", TF_FLOAT); TFE_OpAddInput(op, var_handle, status); // Convert 'value' to a TF_Tensor then a TFE_TensorHandle.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2)