Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for addInputs (0.11 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
Back to top