- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for pushStack (0.12 sec)
-
tensorflow/c/eager/c_api_remote_test_util.cc
for (int i = 0; i < 50; ++i) { handles_task0.push_back(TestMatrixTensorHandle(ctx)); } } const char task1_name[] = "/job:localhost/replica:0/task:1/device:CPU:0"; const char task2_name[] = "/job:localhost/replica:0/task:2/device:CPU:0"; std::vector<TFE_TensorHandle*> handles_task2; for (auto* h_task0 : handles_task0) { handles_task2.push_back(
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Fri Dec 11 22:56:03 UTC 2020 - 9.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_debug.cc
return shape; } shape.reserve(rank); for (int i = 0; i < rank; ++i) { int64_t dim; *status = handle.Dim(i, &dim); if (!status->ok()) { return shape; } shape.push_back(dim); } return shape; } } // namespace extern "C" { TF_CAPI_EXPORT extern TFE_TensorDebugInfo* TFE_TensorHandleTensorDebugInfo( TFE_TensorHandle* h, TF_Status* status) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
TFE_DeleteOp(op); if (TF_GetCode(s) != TF_OK) return; std::vector<TFE_TensorHandle*> unwrapped_outputs; unwrapped_outputs.reserve(op_outputs.size()); for (auto* handle : op_outputs) { unwrapped_outputs.push_back(handle); } for (int i = 0; i < *num_outputs; ++i) { auto logged_tensor = std::make_unique<LoggedTensor>(unwrapped_outputs[i]); outputs[i] = MakeLoggedTensorHandle(context, dev->device_name,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Mar 03 20:47:31 UTC 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
return wrap(unwrap(o)->outputs[i]); } void TF_OutputListPushBack(TF_OutputList* o, TF_AbstractTensor* tensor, TF_Status* s) { unwrap(o)->outputs.push_back(unwrap(tensor)); } void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type, TF_Status* s) { tsl::Set_TF_Status_from_Status( s, unwrap(op)->Reset(op_type,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 9K bytes - Viewed (0)