Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for Garg (0.15 sec)

  1. tensorflow/c/c_api.cc

                .NewSubScope(child_scope_name);
    
        if (dx != nullptr) {
          std::vector<tensorflow::Output> dx_arg = OutputsFromTFOutputs(dx, ny);
          status->status =
              AddSymbolicGradients(scope, y_arg, x_arg, dx_arg, &dy_arg);
        } else {
          status->status = AddSymbolicGradients(scope, y_arg, x_arg, &dy_arg);
        }
    
        // Update g->name_map with the name_map from the scope, which will contain
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental.cc

      if (!status->status.ok()) return nullptr;
    
      if (input_index >= op_def->input_arg_size() || input_index < 0) {
        status->status = tensorflow::errors::InvalidArgument(
            input_index, " out of range for ", op_name);
        return nullptr;
      }
    
      const tensorflow::OpDef_ArgDef& input_arg = op_def->input_arg()[input_index];
    
      if (input_arg.number_attr().empty()) {
        status->status = tensorflow::errors::NotFound(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/next_pluggable_device/c_api.cc

                                                      TF_Status* status) {
      auto* cc_ctx = reinterpret_cast<tensorflow::OpKernelContext*>(ctx);
      const tensorflow::Tensor& arg_tensor = cc_ctx->input(index);
      absl::Status cc_status;
      if (arg_tensor.dtype() != tensorflow::DT_RESOURCE) {
        cc_status = absl::InvalidArgumentError(
            absl::StrCat("Trying to obtain resource handle from Input[", index,
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jan 09 00:52:04 GMT 2024
    - 13.9K bytes
    - Viewed (1)
  4. tensorflow/c/eager/c_api_distributed_test.cc

      CHECK(tensorflow::protobuf::TextFormat::ParseFromString(
          "    signature {"
          "      name: 'AddVariablesFunction'"
          "      input_arg {"
          "        name: 'var'"
          "        type: DT_RESOURCE"
          "      }"
          "      output_arg {"
          "        name: 'sum'"
          "        type: DT_FLOAT"
          "      }"
          "    }"
          "    node_def {"
          "      name: 'read0'"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_test.cc

      EXPECT_EQ(TF_CANCELLED, TF_GetCode(s));
      EXPECT_EQ(string("cancel"), TF_Message(s));
      TF_DeleteStatus(s);
    }
    
    void Deallocator(void* data, size_t, void* arg) {
      tensorflow::cpu_allocator()->DeallocateRaw(data);
      *reinterpret_cast<bool*>(arg) = true;
    }
    
    TEST(CAPI, Tensor) {
      const int num_bytes = 6 * sizeof(float);
      float* values =
          reinterpret_cast<float*>(tensorflow::cpu_allocator()->AllocateRaw(
    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)
  6. tensorflow/c/eager/dlpack.cc

      if (!status->status.ok()) {
        return nullptr;
      }
      return tensor;
    }
    
    // Deleter for DLManagedTensor
    void DLManagedTensorDeleter(DLManagedTensor* arg) {
      TfDlManagedTensorCtx* owner =
          static_cast<TfDlManagedTensorCtx*>(arg->manager_ctx);
      owner->reference.Unref();
      delete owner;
    }
    
    // Converts TF_DATAType to DLPack data type.
    DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) {
    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)
  7. tensorflow/c/eager/c_api.cc

        void (*deallocator)(void* data, size_t len, void* arg),
        void* deallocator_arg, TF_Status* status) {
      tensorflow::Device* device = nullptr;
      tensorflow::EagerContext* context =
          tensorflow::ContextFromInterface(tensorflow::unwrap(ctx));
      status->status = context->FindDeviceFromName(device_name, &device);
      if (!status->status.ok()) {
        deallocator(data, len, deallocator_arg);
        status->status =
    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)
Back to top