- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for TFE_OpAddInput (0.18 sec)
-
tensorflow/c/eager/c_api_test_util.cc
TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, "AddV2", status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(op, a, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(op, b, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TF_DeleteStatus(status); TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a));
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
int num_retvals = 1; for (auto s : state) { TFE_OpReset(matmul, "MatMul", nullptr, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(matmul, m, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(matmul, m, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_Execute(matmul, &retvals[0], &num_retvals, status);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api_cluster_test.cc
TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpSetAttrType(op, "dtype", TF_FLOAT); TFE_OpAddInput(op, var, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); int num_retvals = 1; TFE_Execute(op, out_value, &num_retvals, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); TFE_Op* func = TFE_NewOp(ctx, "AddVariablesFunction", status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); TFE_OpAddInput(func, packed_handle, status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); if (remote) { TFE_OpSetDevice(func, task1_name, status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 23.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
TFE_NewOp(context.get(), "TPUReplicatedOutput", status.get()), TFE_DeleteOp); if (TF_GetCode(status.get()) != TF_OK) return; TFE_OpSetAttrInt(op.get(), "num_replicas", 1); TFE_OpAddInput(op.get(), value_one.get(), status.get()); if (TF_GetCode(status.get()) != TF_OK) return; TFE_OpSetDevice(op.get(), device_name, status.get()); if (TF_GetCode(status.get()) != TF_OK) return;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Tue Aug 06 23:56:17 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TF_DeleteTensor(t); TFE_Op* op = TFE_NewOp(ctx, "ident", status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TFE_OpAddInput(op, h, status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); std::vector<TFE_TensorHandle*> result; result.push_back(nullptr); int num_retvals = 1;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 03 03:14:26 UTC 2023 - 31.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api.h
TF_CAPI_EXPORT extern const char* TFE_OpGetDevice(const TFE_Op* op, TF_Status* status); TF_CAPI_EXPORT extern void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status); TF_CAPI_EXPORT extern void TFE_OpAddInputList(TFE_Op* op,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
if (TF_GetCode(status) != TF_OK) return; } TFE_OpAddAttrs(op_.get(), attributes); for (int input_index = 0; input_index < inputs.size(); ++input_index) { TFE_OpAddInput(op_.get(), inputs[input_index], status); if (TF_GetCode(status) != TF_OK) return; } std::vector<TFE_TensorHandle*> unwrapped_results(expected_max_outputs); int real_num_outputs = expected_max_outputs;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
status->status = tensorflow::unwrap(op)->SetDeviceName(device_name); } const char* TFE_OpGetDevice(const TFE_Op* op, TF_Status* status) { return tensorflow::unwrap(op)->DeviceName().c_str(); } void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) { status->status = tensorflow::unwrap(op)->AddInput(tensorflow::unwrap(input)); } void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)