Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for HasXLAKernel (0.11 sec)

  1. tensorflow/compiler/jit/compilability_check_util.cc

      IsCompilableNode(node, lib_runtime, &stack_trace,
                       /*encapsulating_function=*/nullptr, &uncompilable_nodes);
      return uncompilable_nodes;
    }
    
    bool RecursiveCompilabilityChecker::HasXLAKernel(
        const Node& node, string* uncompilable_reason) const {
      // There is a SymbolicGradient kernel on the XLA_JIT device, but the gradient
      // is really a kind of function call and will be handled by
    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/compiler/jit/compilability_check_util.h

        auto is_variant = [](DataType dtype) { return dtype == DT_VARIANT; };
        return absl::c_any_of(node.input_types(), is_variant) ||
               absl::c_any_of(node.output_types(), is_variant);
      }
    
      bool HasXLAKernel(const Node& node,
                        string* uncompilable_reason = nullptr) const;
    
      static void MaybeMarkUncompilableNode(
          const absl::string_view reason,
          const std::vector<StackFrameView>& stack_trace,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 14.9K bytes
    - Viewed (0)
Back to top