Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 41 for init (0.2 sec)

  1. tensorflow/c/experimental/gradients/array_grad.cc

     public:
      Status Compute(AbstractContext* ctx,
                     absl::Span<AbstractTensorHandle* const> grad_outputs,
                     absl::Span<AbstractTensorHandle*> grad_inputs) override {
        for (int i = 0; i < grad_outputs.size(); i++) {
          auto grad_input = grad_outputs[i];
          // TODO(srbs): Should we add a copy contructor to AbstractTensorHandle
          // that takes care of this similar to `Tensor`?
    C++
    - Registered: Tue Apr 09 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 1.6K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/gradients/math_grad_test.cc

                             {12, 15, 18, 12, 15, 18, 12, 15, 18},
                             {6, 6, 6, 15, 15, 15, 24, 24, 24},
                             {6, 15, 24, 6, 15, 24, 6, 15, 24}};
    
      for (int i{}; i < 4; ++i) {
        bool transpose_a = transpose_a_vals[i];
        bool transpose_b = transpose_b_vals[i];
        Model MatMulModel =
            [transpose_a, transpose_b](
                AbstractContext* ctx,
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Thu Apr 13 17:32:14 GMT 2023
    - 16.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients.cc

    }
    Status SetAttrStringList(AbstractOperation* op_, const char* attr_name,
                             const void* const* values, const size_t* lengths,
                             int num_values, ForwardOperation* forward_op_) {
      std::vector<StringPiece> v(num_values);
      for (int i = 0; i < num_values; ++i) {
        v[i] = StringPiece(static_cast<const char*>(values[i]), lengths[i]);
      }
      forward_op_->attrs.Set(attr_name, v);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/dlpack.cc

    // data.
    bool IsValidStrideCompactRowMajorData(int64_t* shape_arr, int64_t* stride_arr,
                                          int ndim) {
      bool valid = true;
      int64_t expected_stride = 1;
      for (int i = ndim - 1; i >= 0; --i) {
        // Empty tensors are always compact regardless of strides.
        if (shape_arr[i] == 0) return true;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/gradients/not_differentiable.cc

    namespace tensorflow {
    namespace gradients {
    Status NotDifferentiableGradientFunction::Compute(
        AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs,
        absl::Span<AbstractTensorHandle*> grad_inputs) {
      for (int i = 0; i < grad_inputs.size(); i++) {
        grad_inputs[i] = nullptr;
      }
      return OkStatus();
    }
    
    Status RegisterNotDifferentiable(GradientRegistry* registry, const string& op) {
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Jun 15 01:15:58 GMT 2022
    - 1.3K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/filesystem/modular_filesystem_registration.cc

      if (ops == nullptr)
        return errors::FailedPrecondition(
            "Trying to register filesystem without operations");
    
      if (ops->init == nullptr)
        return errors::FailedPrecondition(
            "Trying to register filesystem without `init` operation");
    
      if (ops->cleanup == nullptr)
        return errors::FailedPrecondition(
            "Trying to register filesystem without `cleanup` operation");
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 07 22:08:43 GMT 2023
    - 12.8K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api.cc

    TF_CAPI_EXPORT extern void TFE_ContextSetServerDefWithTimeout(
        TFE_Context* ctx, int keep_alive_secs, const void* proto, size_t proto_len,
        int64_t init_timeout_in_ms, TF_Status* status,
        bool clear_existing_contexts) {
      TFE_ContextSetServerDefWithTimeoutAndRetries(
          ctx, keep_alive_secs, proto, proto_len, init_timeout_in_ms,
          /*retries=*/0, status, clear_existing_contexts);
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  8. tensorflow/c/c_api_test.cc

      }
    }
    
    TEST_F(CApiAttributesTest, EmptyList) {
      auto desc = init("list(int)");
      TF_SetAttrIntList(desc, "v", nullptr, 0);
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_TF_META("v", 0, TF_ATTR_INT, -1);
    }
    
    TEST_F(CApiAttributesTest, Names) {
      auto desc = init("string");
      TF_SetAttrString(desc, "v", "bunny", 5);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  9. tensorflow/c/eager/c_api_test_util.cc

      }
      return server_def;
    }
    
    tensorflow::ServerDef GetServerDef(int num_tasks) {
      return GetServerDef("localhost", num_tasks);
    }
    
    tensorflow::ServerDef GetMultiClientServerDef(const std::string& job_name,
                                                  int num_tasks,
                                                  int num_virtual_gpus) {
      tensorflow::ServerDef server_def;
      server_def.set_protocol("grpc");
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  10. tensorflow/c/c_api_function.cc

        const TF_Graph* fn_body, const char* fn_name, int ninputs,
        const TF_Output* inputs, std::vector<OutputTensor>* input_tensors,
        std::unordered_map<const Node*, std::vector<int>>* input_nodes)
        TF_EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) {
      input_tensors->reserve(ninputs);
      for (int i = 0; i < ninputs; ++i) {
        Node* node = inputs[i].oper ? &inputs[i].oper->node : nullptr;
        int idx = inputs[i].index;
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
Back to top