- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for HasXLAKernel (0.16 sec)
-
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) -
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)