- Sort Score
- Num 10 results
- Language All
Results 1 - 4 of 4 for TFE_OpGetFlatInput (0.12 seconds)
-
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
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 22.8K bytes - Click Count (0) -
tensorflow/c/eager/custom_device_testutil.cc
if (TF_GetCode(s) != TF_OK) return; int num_inputs = TFE_OpGetFlatInputCount(original_op, s); if (TF_GetCode(s) != TF_OK) return; for (int j = 0; j < num_inputs; ++j) { TFE_TensorHandle* input = TFE_OpGetFlatInput(original_op, j, s); if (TF_GetCode(s) != TF_OK) return; const char* input_device = TFE_TensorHandleDeviceName(input, s); if (TF_GetCode(s) != TF_OK) return; if (dev->device_name == input_device) {
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat Dec 20 23:30:21 GMT 2025 - 8.4K bytes - Click Count (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]); }
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Fri Nov 07 05:55:21 GMT 2025 - 43.9K bytes - Click Count (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); }
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Thu Oct 09 05:56:18 GMT 2025 - 94.6K bytes - Click Count (0)