- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 20 for id (0.16 sec)
-
tensorflow/c/eager/c_api_experimental_test.cc
TF_OperationDescription* id_descr = TF_NewOperation(function_graph, "Identity", "id"); TF_SetAttrType(id_descr, "T", TF_INT32); TF_AddInput(id_descr, {arg, 0}); TF_Operation* id = TF_FinishOperation(id_descr, status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TF_Output input{arg, 0}; TF_Output output{id, 0}; TF_Function* fn =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/eager/c_api_unified_experimental_test.cc
// Test that full type information can be accessed. auto outs = unwrap(add_outputs); auto h = outs->outputs[0]; ASSERT_NE(h, nullptr); ASSERT_EQ(h->FullType().type_id(), TFT_UNSET); ASSERT_EQ(unwrap(inputs[0])->FullType().type_id(), TFT_UNSET); // Clean up operation and inputs. TF_DeleteAbstractOp(add_op); TF_AbstractFunction* func = TF_FinalizeFunction(graph_ctx, add_outputs, status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
tensorflow::DeviceNameUtils::ParseFullName(device_name, &parsed_name); std::string device_type = parsed_name.type; int device_id = 0; if (parsed_name.has_id) { device_id = parsed_name.id; } ctx.device_id = device_id; if (device_type == "CPU") { ctx.device_type = DLDeviceType::kDLCPU; } else if (device_type == "GPU") { #if TENSORFLOW_USE_ROCM
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
for (int i = 0; i < target_tensor_ids.size(); ++i) { int64_t target_id = target_tensor_ids[i]; if (sources_set.find(target_id) != sources_set.end()) { auto tensor = targets[i]; sources_that_are_targets.insert( std::make_pair(target_id, TapeTensor(tensor))); } } TF_RETURN_IF_ERROR(GradientTape::ComputeGradient(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
while (!tensor_stack.empty()) { int64_t tensor_id = tensor_stack.back(); tensor_stack.pop_back(); auto op_id_it = tensor_tape.find(tensor_id); if (op_id_it == tensor_tape.end()) { continue; } int64_t op_id = op_id_it->second; auto op_it = op_tape->find(op_id); auto result_op_it = result.op_tape.find(op_id); if (op_id == -1 || op_it == op_tape->end() ||
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 02 12:40:29 GMT 2024 - 47.2K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental.h
bool use_tfrt); // Returns the context_id from the EagerContext which is used by the // EagerService to maintain consistency between client and worker. The // context_id is initialized with a dummy value and is later set when the worker // is initialized (either locally or remotely). The context_id can change during // the process lifetime although this should cause the worker to be
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
TF_Status* status); // On success, enqueues `tensor` into a TF-managed FifoQueue given by // `tensor_id`, associated with `session`. There must be a graph node named // "fifo_queue_enqueue_<tensor_id>", to be executed by this API call. It reads // from a placeholder node "arg_tensor_enqueue_<tensor_id>". //
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
} return ret; } TF_Tensor* TF_DequeueNamedTensor(TF_Session* session, int tensor_id, TF_Status* status) { assert(session); { tensorflow::mutex_lock c(session->graph->mu); VLOG(1) << "Dequeuing named tensor with id " << tensor_id << ", with input graph: " << session->graph->graph.ToGraphDefDebug().DebugString(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
auto* task_iter = static_cast<const tensorflow::CoordinatedTask*>(tasks.data); for (size_t i = 0; i < tasks.length; ++i) { task_vec[i].set_job_name(task_iter->job_name()); task_vec[i].set_task_id(task_iter->task_id()); ++task_iter; } auto results = coord_agent->GetTaskState(task_vec); if (!results.ok()) { status->status = results.status(); return; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/c_api_experimental_test.cc
ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); EXPECT_EQ(assign, 1); int id = TF_OpIsStateful("Identity", status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); EXPECT_EQ(id, 0); } class ShapeInferenceTest : public ::testing::Test { protected: ShapeInferenceTest()
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jan 17 22:27:52 GMT 2023 - 13.1K bytes - Viewed (1)