- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 19 for insert (0.36 sec)
-
tensorflow/c/c_api_experimental.cc
TF_DataType value) { auto iter = builder->attr_names.insert(attr_name).first; builder->Set(*iter, static_cast<tensorflow::DataType>(value)); } void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder, const char* attr_name, const TF_DataType* values, int num_values) { auto iter = builder->attr_names.insert(attr_name).first;
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_cluster_test.cc
TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpSetAttrType(op, "dtype", TF_FLOAT); TFE_OpAddInput(op, var, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); int num_retvals = 1; TFE_Execute(op, out_value, &num_retvals, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_DeleteOp(op);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
auto iter = registry_.find(op_name); if (iter != registry_.end()) { const string error_msg = "Gradient already exists for op: " + op_name + "."; return errors::AlreadyExists(error_msg); } registry_.insert({op_name, gradient_function_factory}); return absl::OkStatus(); } Status GradientRegistry::Lookup( const ForwardOperation& op, std::unique_ptr<GradientFunction>* gradient_function) const {
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/experimental/gradients/math_grad_test.cc
status_ = TestScalarTensorHandle<float, TF_FLOAT>( immediate_execution_ctx_.get(), 2.0f, &x_raw); ASSERT_EQ(errors::OK, status_.code()) << status_.message(); x.reset(x_raw); } status_ = registry_.Register("Exp", ExpRegisterer); ASSERT_EQ(errors::OK, status_.code()) << status_.message(); ASSERT_NO_FATAL_FAILURE(CompareNumericalAndAutodiffGradients( ExpModel, BuildGradModel(ExpModel, registry_),
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Apr 13 17:32:14 GMT 2023 - 16.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
{"subdiv_offsets", std::vector<int>()}}, /*dep=*/{"assert"}}, }); TF_ASSERT_OK(ContextFromInterface(unwrap(context.get())) ->AddFunctionDef(assert_and_collective)); std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> call_op( TFE_NewOp(context.get(), "AssertAndCollective", status.get()), TFE_DeleteOp); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
" into function '", fn_name, "'"); input_tensors->emplace_back(node, idx); const auto& iter = input_nodes->find(node); if (iter == input_nodes->end()) { input_nodes->insert({node, {idx}}); } else { auto& indices = iter->second; if (std::find(indices.begin(), indices.end(), idx) != indices.end()) { return InvalidArgument("TF_Output ", node->name(), ":", idx,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/eager/custom_device_test.cc
TFE_TensorHandle* hcpu = TestMatrixTensorHandle(context); ASSERT_FALSE(arrived); TFE_TensorHandle* hdevice = TFE_TensorHandleCopyToDevice(hcpu, context, name, status.get()); ASSERT_TRUE(arrived); ASSERT_FALSE(executed); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> matmul( MatMulOp(context, hcpu, hdevice), TFE_DeleteOp);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
TF_OperationDescription* desc = TF_NewOperation(graph, "Const", name); TF_SetAttrTensor(desc, "value", t, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); TF_SetAttrType(desc, "dtype", TF_TensorType(t)); *op = TF_FinishOperation(desc, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); ASSERT_NE(*op, nullptr); } TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/eager/c_api_test_util.cc
tensorflow::JobDef* job_def = cluster_def->add_job(); job_def->set_name(job_name); for (int i = 0; i < num_tasks; i++) { int port = tensorflow::testing::PickUnusedPortOrDie(); job_def->mutable_tasks()->insert( {i, tensorflow::strings::StrCat("localhost:", port)}); LOG(INFO) << "Picked test port: " << port << " for job: " << job_name << ", task: " << i; } return server_def; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/eager/c_api_distributed_test.cc
ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); TFE_OpAddInput(func, var_handle, status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); TFE_TensorHandle* retvals[1] = {nullptr}; int num_retvals = 1; TFE_Execute(func, &retvals[0], &num_retvals, status); if (inject_error) { ASSERT_EQ(TF_INTERNAL, TF_GetCode(status)) << TF_Message(status); } else {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0)