- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for addInputs (0.11 sec)
-
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/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/translate/export_graphdef.cc
if (n->IsOp()) { NodeDef* node_def = n->mutable_def(); node_def->clear_input(); for (const Edge* e : n->in_edges()) { Node* src = e->src(); if (src->IsOp()) { Graph::AddInput(node_def, src->name(), e->src_output()); } } } } } absl::StatusOr<std::unique_ptr<Graph>> Exporter::Convert( const GraphExportConfig& configs, const Dialect* tf_dialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
if (n->IsOp()) { NodeDef* node_def = n->mutable_def(); node_def->clear_input(); for (const Edge* e : n->in_edges()) { Node* src = e->src(); if (src->IsOp()) { Graph::AddInput(node_def, src->name(), e->src_output()); } } } } } absl::StatusOr<std::unique_ptr<Graph>> Exporter::Convert( const GraphExportConfig& configs, const Dialect* tf_dialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
return tensorflow::unwrap(op)->DeviceName().c_str(); } void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) { status->status = tensorflow::unwrap(op)->AddInput(tensorflow::unwrap(input)); } void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs, TF_Status* status) { status->status = tensorflow::unwrap(op)->AddInputList(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0)