Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for add_inputs (0.2 sec)

  1. 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: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.cc

      forward_op_->op_name = op;
      forward_op_->attrs.Reset(op);
      return op_->Reset(op, raw_device_name);
    }
    absl::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();
    }
    absl::Status AddInputList(AbstractOperation* op_,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  3. 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: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      absl::Status SetDeviceName(const char* name) override {
        // TODO(srbs): Implement this.
        device_name_ = name;
        return absl::OkStatus();
      }
    
      absl::Status AddInput(AbstractTensorHandle* input) override {
        GraphTensor* t = dyn_cast<GraphTensor>(input);
        if (!t) {
          return tensorflow::errors::InvalidArgument(
              "Unable to cast input to GraphTensor");
        }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  5. 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: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top