Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for GetFunctionLibraryDefinition (0.51 sec)

  1. tensorflow/compiler/jit/xla_compiler_options_util.cc

      if (stream != nullptr) {
        options.device_ordinal = stream->parent()->device_ordinal();
      }
      options.device_type = xla_device_compiler.device_type();
      options.flib_def = function_library.GetFunctionLibraryDefinition();
      options.graph_def_version = function_library.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: Wed Feb 21 09:53:30 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfrt/tests/saved_model/saved_model_test.cc

      // The module contains an XLA function, as well as a while body and a while
      // condition within the XLA function.
      EXPECT_EQ(fallback_state->process_function_library_runtime()
                    .GetFunctionLibraryDefinition()
                    ->num_functions(),
                3);
    }
    
    TEST(SavedModelTest, ConvertTfMlirToBefExportingXlaReduceWindow) {
      std::string saved_model_mlir_path = tensorflow::GetDataDependencyFilepath(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Oct 13 01:17:29 UTC 2023
    - 9K bytes
    - Viewed (0)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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