Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 17 for if (0.12 sec)

  1. CONTRIBUTING.md

      `comp:` etc.  At this stage we check if the PR is valid and meets certain
      quality requirements. For example, we check if the CLA is signed, PR has
      sufficient description, if applicable unit tests are added, if it is a
      reasonable contribution (meaning it is not a single liner cosmetic PR).
    
    **2. Valid?**
    
    -   If the PR passes all the quality checks then we go ahead and assign a
        reviewer.
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Thu Mar 21 11:45:51 GMT 2024
    - 15.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.cc

      AbstractOperationPtr op(ctx->CreateOperation());
      TF_RETURN_IF_ERROR(op->Reset("ZerosLike", /*raw_device_name=*/nullptr));
      if (isa<tracing::TracingOperation>(op.get())) {
        TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingOperation>(op.get())->SetOpName(
            absl::StrCat("ZerosLike", ToId(t)).c_str()));
      }
      TF_RETURN_IF_ERROR(op->AddInput(t));
      int num_outputs = 1;
    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)
  3. tensorflow/c/eager/dlpack.cc

      std::string device_type = parsed_name.type;
      int device_id = 0;
      if (parsed_name.has_id) {
        device_id = parsed_name.id;
      }
    
      ctx.device_id = device_id;
      if (device_type == "CPU") {
        ctx.device_type = DLDeviceType::kDLCPU;
      } else if (device_type == "GPU") {
    #if TENSORFLOW_USE_ROCM
        ctx.device_type = DLDeviceType::kDLROCM;
    #else
    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)
  4. tensorflow/c/eager/c_api_experimental.h

    // is for performance optimization by reusing an exiting unused op rather than
    // creating a new op every time. If `raw_device_name` is `NULL` or empty, it
    // does not set the device name. If it's not `NULL`, then it attempts to parse
    // and set the device name. It's effectively `TFE_OpSetDevice`, but it is faster
    // than separately calling it because if the existing op has the same
    // `raw_device_name`, it skips parsing and just leave as it is.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  5. tensorflow/c/BUILD

    load("@bazel_skylib//lib:selects.bzl", "selects")
    load("@local_config_tensorrt//:build_defs.bzl", "if_tensorrt")
    load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
    load(
        "//tensorflow:tensorflow.bzl",
        "check_deps",
        "if_google",
        "if_not_mobile",
        "tf_cc_test",
        "tf_copts",
        "tf_cuda_library",
        "tf_custom_op_library",
        "tf_kernel_library",
    )
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 27 18:00:18 GMT 2024
    - 30.3K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/gradients/math_grad.cc

        // Get transpose attrs
        bool t_a;
        TF_RETURN_IF_ERROR(forward_attrs_.Get("transpose_a", &t_a));
    
        bool t_b;
        TF_RETURN_IF_ERROR(forward_attrs_.Get("transpose_b", &t_b));
    
        // Conj each input
        AbstractTensorHandle* conj_output;
        std::string name = "Conj_A_MatMul_Grad";
        TF_RETURN_IF_ERROR(
            SafeConj(ctx, forward_inputs_[0], &conj_output, name.c_str()));
    
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 15.2K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api.cc

      if (!status->status.ok()) return nullptr;
    
      return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor));
    }
    
    void TFE_DeleteTensorHandle(TFE_TensorHandle* h) {
      if (h == nullptr) return;
    
      tensorflow::profiler::TraceMe activity(
          "TFE_DeleteTensorHandle", tensorflow::profiler::TraceMeLevel::kInfo);
      if (h) {
        tensorflow::unwrap(h)->Unref();
      }
    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/eager/c_api_unified_experimental_graph.cc

        TracingOperationPtr operation(CreateOperation());
        TF_RETURN_IF_ERROR(operation->Reset("Placeholder", nullptr));
        TF_RETURN_IF_ERROR(
            operation->SetOpName(absl::StrCat("_input_", inputs_.size()).c_str()));
        TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype));
        if (!shape.unknown_rank()) {
          TF_RETURN_IF_ERROR(operation->SetAttrShape(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  9. tensorflow/c/eager/c_api_experimental.cc

        dimvec[i] = static_cast<int64_t>(dims[i]);
      }
    
      if (ctx == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid Context");
        return nullptr;
      }
    
      tensorflow::AbstractTensorInterface* t =
          tensorflow::unwrap(ctx)->CreateTensor(
              static_cast<tensorflow::DataType>(dtype), dimvec);
    
      if (t == nullptr) {
        status->status =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
  10. tensorflow/c/eager/c_api_test_util.cc

      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_OpSetAttrType(op, "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(op, "shape", {}, 0, status);
      TFE_OpSetAttrString(op, "container", "localhost", 0);
      TFE_OpSetAttrString(op, "shared_name", "", 0);
      if (!device_name.empty()) {
        TFE_OpSetDevice(op, device_name.c_str(), status);
      }
      if (TF_GetCode(status) != TF_OK) return nullptr;
    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)
Back to top