Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for NameAndAttrsFromFunctionCall (0.4 sec)

  1. tensorflow/compiler/jit/force_xla_constants_on_host_pass.cc

          const FunctionBody* fbody = nullptr;
          std::vector<int> constant_arg_indices;
          std::vector<int> resource_arg_indices;
    
          NameAttrList function;
          TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(node->def(), &function));
    
          // Force all constants to be on the host memory.
          TF_RETURN_IF_ERROR(GetBodyAndConstantsAndResources(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 2.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_kernel_creator.cc

      XlaOpRegistry::RegisterCompilationKernels();
    
      // Get function body, constant args, and resource args.
      NameAttrList function;
      TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(node_def, &function));
      const FunctionBody* fbody = nullptr;
      std::vector<int> constant_arg_indices;
      std::vector<int> resource_arg_indices;
      TF_RETURN_IF_ERROR(GetBodyAndConstantsAndResources(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 22:24:01 UTC 2023
    - 4.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/compilability_check_util.cc

                << ".";
        return false;
      }
    
      FunctionLibraryRuntime::Handle handle;
      Status s;
      NameAttrList function;
      s = NameAndAttrsFromFunctionCall(call_def, &function);
      if (s.ok()) {
        s = lib_runtime->Instantiate(function.name(), AttrSlice(&function.attr()),
                                     &handle);
      }
      if (!s.ok()) {
    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