Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for GetBodyAndConstantsAndResources (0.55 sec)

  1. tensorflow/compiler/jit/force_xla_constants_on_host_pass.cc

          NameAttrList function;
          TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(node->def(), &function));
    
          // Force all constants to be on the host memory.
          TF_RETURN_IF_ERROR(GetBodyAndConstantsAndResources(
              flr, function, &fbody, &constant_arg_indices, &resource_arg_indices));
          VLOG(3) << "Found constant arg indices: "
                  << absl::StrJoin(constant_arg_indices, ", ");
    
    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

      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(
          flr, function, &fbody, &constant_arg_indices, &resource_arg_indices));
    
      MemoryTypeVector input_memory_types =
          GetInputMemoryTypes(fbody, constant_arg_indices, resource_arg_indices);
    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.h

    // `fbody` is owned by `flr`.
    // `constant_arg_indices` and `resource_arg_indices` should be empty vector.
    // They are sorted in ascending order on this function's return.
    Status GetBodyAndConstantsAndResources(FunctionLibraryRuntime* flr,
                                           const NameAttrList& function,
                                           const FunctionBody** fbody,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 14.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/get_compiler_ir.cc

        CompilerArgSource compiler_arg_source) {
      const FunctionBody* fbody = nullptr;
      std::vector<int> constant_arg_indices;
      std::vector<int> resource_arg_indices;
      TF_RETURN_IF_ERROR(GetBodyAndConstantsAndResources(
          flr, function, &fbody, &constant_arg_indices, &resource_arg_indices));
      if (dev == nullptr && !resource_arg_indices.empty()) {
        return absl::InternalError(
    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/compilability_check_util.cc

      return it != node.attr().end() && it->second.b();
    }
    
    bool CanCreateXlaKernel(const NodeDef& node_def) {
      return HasBoolAttr(node_def, kXlaMustCompileAttr);
    }
    
    Status GetBodyAndConstantsAndResources(FunctionLibraryRuntime* flr,
                                           const NameAttrList& function,
                                           const FunctionBody** fbody,
    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