Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for TF_StringView (0.23 sec)

  1. tensorflow/c/kernels.cc

      return res;
    }
    #endif
    
    TF_StringView TF_GetOpKernelName(TF_OpKernelContext* ctx) {
      auto cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
      TF_StringView opkernel_name_sv;
      opkernel_name_sv.data = cc_ctx->op_kernel().name().data();
      opkernel_name_sv.len = cc_ctx->op_kernel().name().length();
      return opkernel_name_sv;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  2. tensorflow/c/kernels_experimental.h

    //   1. Cannot allocate a new temporary variable
    //   2. Calling plugin allocator failed
    TF_CAPI_EXPORT extern void TF_TemporaryVariable(
        TF_OpKernelContext* ctx, TF_DataType dtype, const int64_t* dims,
        int num_dims, TF_StringView* var_name,
        void (*plugin_allocator)(TF_OpKernelContext*, TF_Tensor*, TF_DataType,
                                 const int64_t*, int, TF_Status*),
        TF_Status* tf_status);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Aug 07 14:44:39 UTC 2023
    - 9.4K bytes
    - Viewed (0)
  3. tensorflow/c/kernels.h

    // has the same lifetime as the OpKernel.
    TF_CAPI_EXPORT extern TF_StringView TF_GetOpKernelName(TF_OpKernelContext* ctx);
    
    // Returns the default container of the resource manager in OpKernelContext.
    //
    // The returned TF_StringView's underlying string is owned by the OpKernel and
    // has the same lifetime as the OpKernel.
    TF_CAPI_EXPORT extern TF_StringView TF_GetResourceMgrDefaultContainerName(
        TF_OpKernelContext* ctx);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 09 22:46:22 UTC 2024
    - 24.6K bytes
    - Viewed (0)
  4. tensorflow/c/kernels/histogram_summary_op.cc

    // kernel computation.
    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)
  5. tensorflow/c/kernels_experimental.cc

      }
      return var_name;
    }
    
    void TF_TemporaryVariable(TF_OpKernelContext* ctx, TF_DataType dtype,
                              const int64_t* dims, int num_dims,
                              TF_StringView* var_name,
                              void (*allocFunc)(TF_OpKernelContext*, TF_Tensor*,
                                                TF_DataType, const int64_t*, int,
                                                TF_Status*),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 06:12:29 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  6. tensorflow/c/kernels_test.cc

      TF_OpKernelConstruction_GetAttrType(ctx, "SomeDataTypeAttr", &type, status);
      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);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
  7. tensorflow/c/c_api.h

    // Used to return strings across the C API. The caller does not take ownership
    // of the underlying data pointer and is not responsible for freeing it.
    typedef struct TF_StringView {
      const char* data;
      size_t len;
    } TF_StringView;
    
    // --------------------------------------------------------------------------
    // TF_SessionOptions holds options that can be passed during session creation.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
Back to top