- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TFE_OpGetFlatInputCount (0.5 sec)
-
tensorflow/c/eager/c_api.h
// 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 // particular named input list, which may only be part of the op's inputs). TF_CAPI_EXPORT extern int TFE_OpGetFlatInputCount(const TFE_Op* op, TF_Status* status);
Registered: Sun Jun 16 05:45:23 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
NamedParallelDevice* named_device = reinterpret_cast<NamedParallelDevice*>(device_info); std::vector<MaybeParallelTensorUnowned> typed_inputs; 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);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 29 22:05:31 UTC 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
{reinterpret_cast<tensorflow::AbstractTensorHandle**>( tensorflow::unwrap(inputs)), 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,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0)