Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for GetFunctionLibraryDefinition (0.45 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc

      std::shared_ptr<const tensorflow::NodeProperties> props;
      absl::Status status = tensorflow::NodeProperties::CreateFromNodeDef(
          *nodedef_or.value(),
          params_.function_library->GetFunctionLibraryDefinition(), &props);
      if (!status.ok()) {
        return op_->emitRemark()
               << "failed to create NodeProperties: " << status.ToString();
      }
      tensorflow::OpKernel* op_kernel_raw;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 18.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/get_compiler_ir.cc

        options = GenerateCompilerOptionsForTfrtTpu(*xla_device_compiler, *flr);
      } else {
        options.device_type = compilation_device_type;
        options.flib_def = flr->GetFunctionLibraryDefinition();
        options.graph_def_version = flr->graph_def_version();
        options.allow_cpu_custom_calls =
            (platform_info.platform_id() == se::host::kHostPlatformId);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_cluster_util.cc

    CallTargetListTy GetCallTargetListFromNode(
        const Node& n, FunctionLibraryRuntime* lib_runtime) {
      const FunctionLibraryDefinition& flib_def =
          *lib_runtime->GetFunctionLibraryDefinition();
      if (flib_def.Find(n.type_string())) {
        NameAttrList callee;
        callee.set_name(n.type_string());
        *callee.mutable_attr() = n.def().attr();
        return {callee};
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  4. tensorflow/c/kernels.cc

        TF_Status* status) {
      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
      auto fdef_lib =
          cc_ctx->function_library()->GetFunctionLibraryDefinition()->ToProto();
      auto cc_status =
          tensorflow::MessageToBuffer(fdef_lib, serialized_function_def_library);
      tensorflow::Set_TF_Status_from_Status(status, cc_status);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/compilability_check_util.cc

        LogNotCompilable(node, uncompilable_reason);
        return false;
      }
    
      string uncompilable_reason;
      if (IsFunctionCall(*lib_runtime->GetFunctionLibraryDefinition(), node)) {
        if (!IsCompilableCall(node.def(), lib_runtime, stack_trace,
                              encapsulating_function, uncompilable_nodes)) {
          LogNotCompilable(node, "unsupported function");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 30.3K bytes
    - Viewed (0)
Back to top