- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for add_input (0.15 sec)
-
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/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/eager/gradients.cc
forward_op_->op_name = op; forward_op_->attrs.Reset(op); return op_->Reset(op, raw_device_name); } Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input, ForwardOperation* forward_op_) { TF_RETURN_IF_ERROR(op_->AddInput(input)); forward_op_->inputs.push_back(input); return absl::OkStatus(); } Status AddInputList(AbstractOperation* op_,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
TF_Status* s, const char* name, TF_Operation** op, bool check) { TF_OperationDescription* desc = TF_NewOperation(graph, "AddN", name); TF_Output add_inputs[2] = {{l, 0}, {r, 0}}; TF_AddInputList(desc, add_inputs, 2); *op = TF_FinishOperation(desc, s); if (check) { ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); ASSERT_NE(*op, nullptr); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 15 03:16:52 UTC 2021 - 17.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_xla.mlir
%bias = arith.constant dense<[7.11401462, 7.05456924]> : tensor<2xf32> %q_input= "quantfork.qcast"(%input) : (tensor<1x3x4x3xf32>) -> tensor<1x3x4x3x!quant.uniform<i8:f32, 0.58810077742034317:-128>> %dq_input= "quantfork.dcast"(%q_input) : (tensor<1x3x4x3x!quant.uniform<i8:f32, 0.58810077742034317:-128>>) -> tensor<1x3x4x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
SmallVector<Value, 8> operands_; llvm::StringMap<Attribute> attrs_; std::unique_ptr<OperationState> state_; // This is the index of the next ODS operand that will be added with AddInput // or AddInput; int current_ods_input_ = 0; const tensorflow::OpDef* op_def_ = nullptr; const char* op_name_ = nullptr; string tf_op_type_; // TODO(srbs): Use this. string device_name_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
Status SetDeviceName(const char* name) override { // TODO(srbs): Implement this. device_name_ = name; return absl::OkStatus(); } Status AddInput(AbstractTensorHandle* input) override { GraphTensor* t = dyn_cast<GraphTensor>(input); if (!t) { return tensorflow::errors::InvalidArgument( "Unable to cast input to GraphTensor"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 20:00:09 UTC 2024 - 15.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
// Rewrite pad_op before Convolutioin. if (failed(HandlePad(pad_op, filter_shape[0], block_size))) return; auto pad_input = pad_op.getInput(); if (auto block_arg = mlir::dyn_cast<mlir::BlockArgument>(pad_input)) { // Change on device function type/shape. HandleFuncOp(block_arg.getOwner()->getParentOp()); } } // Handle Conv2D input, stride and filter.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0)