Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 62 of 62 for input_type (0.38 sec)

  1. tensorflow/compiler/jit/compilability_check_util.cc

    #include "tensorflow/core/public/version.h"
    #include "tensorflow/core/util/dump_graph.h"
    
    namespace tensorflow {
    
    namespace {
    
    bool HasResourceInput(const Node& node) {
      return absl::c_count(node.input_types(), DT_RESOURCE) != 0;
    }
    
    void LogNotCompilable(const Node& node, absl::string_view reason = "") {
      VLOG(3) << "Found uncompilable node " << node.name() << " (op "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 30.3K bytes
    - Viewed (0)
  2. tensorflow/c/kernels.cc

      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
      CHECK_GE(index, 0);                     // Crash OK
      CHECK_LT(index, cc_ctx->num_inputs());  // Crash OK
      return static_cast<TF_DataType>(cc_ctx->input_dtype(index));
    }
    
    void TF_SetOutput(TF_OpKernelContext* ctx, int i, const TF_Tensor* tensor,
                      TF_Status* status) {
      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
Back to top