- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TFE_OpGetFlatInput (0.1 sec)
-
tensorflow/c/eager/c_api.h
// // Does not use the operation's definition to determine how many inputs should // be attached. It is intended for use with TFE_OpGetFlatInput to inspect an // already-finalized operation. // // Note that TFE_OpGetFlatInputCount and TFE_OpGetFlatInput operate on a flat // sequence of inputs, unlike TFE_OpGetInputLength (for getting the length of a
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.cc
int num_inputs = TFE_OpGetFlatInputCount(original_op, status); if (TF_GetCode(status) != TF_OK) return; typed_inputs.reserve(num_inputs); for (int i = 0; i < num_inputs; ++i) { TFE_TensorHandle* input = TFE_OpGetFlatInput(original_op, i, status); if (TF_GetCode(status) != TF_OK) return; const char* tensor_handle_device = TFE_TensorHandleDeviceName(input, status); if (TF_GetCode(status) != TF_OK) return;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
static_cast<size_t>(num_inputs)}); } extern int TFE_OpGetFlatInputCount(const TFE_Op* op, TF_Status* status) { return tensorflow::unwrap(op)->GetInputs().size(); } extern TFE_TensorHandle* TFE_OpGetFlatInput(const TFE_Op* op, int index, TF_Status* status) { return tensorflow::wrap(tensorflow::unwrap(op)->GetInputs()[index]); }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
int num_inputs = TFE_OpGetFlatInputCount(other, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); for (int input_index = 0; input_index < num_inputs; ++input_index) { TFE_TensorHandle* input = TFE_OpGetFlatInput(other, input_index, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(ret, input, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(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)