Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TF_OpKernelConstruction_GetName (0.48 sec)

  1. tensorflow/c/kernels/histogram_summary_op.cc

    struct HistogramSummaryOp {
      std::string op_node_name;
    };
    
    void* HistogramSummaryOp_Create(TF_OpKernelConstruction* ctx) {
      HistogramSummaryOp* kernel = new HistogramSummaryOp;
      TF_StringView string_view_name = TF_OpKernelConstruction_GetName(ctx);
      kernel->op_node_name =
          std::string(string_view_name.data, string_view_name.len);
      return kernel;
    }
    
    void HistogramSummaryOp_Delete(void* kernel) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 6.5K bytes
    - Viewed (0)
  2. tensorflow/c/kernels.h

        TF_OpKernelConstruction* ctx, const char* attr_name, TF_Status* status);
    
    // Returns the unique operation name for this OpKernel.
    TF_CAPI_EXPORT extern TF_StringView TF_OpKernelConstruction_GetName(
        TF_OpKernelConstruction* ctx);
    
    // Allocates Tensor for output at given index. Caller takes ownership of
    // returned TF_Tensor and should deallocate it using TF_DeleteTensor(tensor).
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 09 22:46:22 UTC 2024
    - 24.6K bytes
    - Viewed (0)
  3. tensorflow/c/kernels.cc

      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx);
      return cc_ctx->HasAttr(attr_name);
    }
    
    TF_StringView TF_OpKernelConstruction_GetName(TF_OpKernelConstruction* ctx) {
      auto* cc_ctx = reinterpret_cast<tensorflow::OpKernelConstruction*>(ctx);
      TF_StringView string_view_of_name;
      string_view_of_name.data = cc_ctx->def().name().data();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  4. tensorflow/c/kernels_test.cc

      EXPECT_EQ(TF_OK, TF_GetCode(status));
      EXPECT_EQ(TF_FLOAT, type);
      TF_DeleteStatus(status);
    
      // Exercise kernel NodeDef name read
      TF_StringView name_string_view = TF_OpKernelConstruction_GetName(ctx);
      std::string node_name = "SomeNodeName";
      std::string candidate_node_name =
          std::string(name_string_view.data, name_string_view.len);
      EXPECT_EQ(node_name, candidate_node_name);
      return s;
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
Back to top