- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 47 for add_input (0.15 sec)
-
tensorflow/c/kernels/summary_op_test.cc
def.set_device(DEVICE_CPU); AttrValue valuesTypeAttr; SetAttrValue(values->dtype(), &valuesTypeAttr); (*def.mutable_attr())["T"] = valuesTypeAttr; def.add_input(strings::StrCat("input1: ", DataTypeString(tags->dtype()))); def.add_input(strings::StrCat("input2: ", DataTypeString(values->dtype()))); std::unique_ptr<OpKernel> kernel = CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 6.7K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
NodeDef def; def.set_op(op_name); def.set_name(node_name); def.set_device(device_name); def.add_input("input1"); def.add_input("input2"); def.add_input("input3"); def.add_input("input3"); def.add_input("input3"); AttrValue v0; v0.set_type(DataType::DT_INT32); v0.set_i(3); (*def.mutable_attr())["NumInput3"] = v0; AttrValue v1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op_test.cc
SetAttrValue(input_tensor->dtype(), &typeAttr); AttrValue outTypeAttr; SetAttrValue(out_type, &outTypeAttr); (*def.mutable_attr())["T"] = typeAttr; (*def.mutable_attr())["type"] = outTypeAttr; def.add_input( strings::StrCat("input1: ", DataTypeString(input_tensor->dtype()))); std::unique_ptr<OpKernel> kernel = CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
const std::pair<std::string, FunctionDefHelper::AttrValueWrapper>> attrs) { NodeDef node; node.set_name(std::string(name)); node.set_op(std::string(op)); for (const auto& input : inputs) node.add_input(input); for (const auto& attr : attrs) node.mutable_attr()->insert({attr.first, attr.second.proto}); return node; } } // namespace GraphDef DeviceCompilerSerializeTest::GetTestGraph(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/cc/tools/freeze_saved_model.cc
NodeDef* identity_node) { identity_node->set_name(node.name()); identity_node->set_op("Identity"); (*identity_node->mutable_attr())["T"] = node.attr().at("dtype"); identity_node->add_input(node.input(0)); } // Returns the name of the VarHandleOp that provides input (possibly indirectly) // to node with node_name. A typical indirect chain of nodes (that can occur due
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 11 08:05:36 UTC 2023 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
// as later when edges are added to the Node using Graph::AddEdge the // associated NodeDef is not updated. for (int i = 0, e = inst->getNumOperands(); i < e; ++i) { node_def->add_input(); } if (auto attr = inst->getAttrOfType<mlir::StringAttr>("device")) { node_def->set_device(std::string(attr.getValue())); } // Add the node debug info. TF_RETURN_IF_ERROR(ConvertLocation(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
NodeDef node_def; tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op); node_def.set_name(op->Name()); node_def.set_op(op->Name()); for (int i = 0; i < num_inputs; ++i) { node_def.add_input("dummy_input"); } OperationFromInterface(op)->Attrs().FillAttrValueMap(node_def.mutable_attr()); const tensorflow::OpRegistrationData* op_reg_data; status->status =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
Node* outside_compilation_node = lifted_arg_nodes_and_outside_compilation_nodes[i - original_arg_count] .second; node_def.add_input(absl::StrCat(outside_compilation_node->name(), ":", 0)); } TF_ASSIGN_OR_RETURN(n, ReplaceNode(g, n, node_def)); // Add edges from outside compilation nodes to call node.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/c/experimental/ops/math_ops.cc
AbstractOperationPtr op_ptr(ctx->CreateOperation()); TF_RETURN_IF_ERROR(op_ptr->Reset("Mul", raw_device_name)); TF_RETURN_IF_ERROR(MaybeSetOpName(op_ptr.get(), name)); TF_RETURN_IF_ERROR(op_ptr->AddInput(x)); TF_RETURN_IF_ERROR(op_ptr->AddInput(y)); int num_retvals = 1; return op_ptr->Execute(absl::MakeSpan(z, 1), &num_retvals); } // Op: Conj() // Summary: Returns the complex conjugate of a complex number. // // Description:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 12.2K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/golden/testing_ops.cc.golden
TF_RETURN_IF_ERROR(op_ptr->Reset("AccumulatorApplyGradient", raw_device_name)); TF_RETURN_IF_ERROR(MaybeSetOpName(op_ptr.get(), name)); TF_RETURN_IF_ERROR(op_ptr->AddInput(handle)); TF_RETURN_IF_ERROR(op_ptr->AddInput(local_step)); TF_RETURN_IF_ERROR(op_ptr->AddInput(gradient)); int num_retvals = 0; std::vector<AbstractTensorHandle*> dummy_outputs; return op_ptr->Execute(absl::MakeSpan(dummy_outputs), &num_retvals); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 16 19:04:03 UTC 2023 - 6.5K bytes - Viewed (0)