Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for CanCreateKernel (0.13 sec)

  1. tensorflow/compiler/jit/xla_kernel_creator.h

     public:
      // Given a NodeDef 'node_def' and the function library runtime 'flr', returns
      // true if 'node_def' is a call to a compilable function defined in 'flr',
      // with the kXlaCompileAttr set.
      bool CanCreateKernel(
          const FunctionLibraryRuntime& flr,
          const std::shared_ptr<const NodeProperties>& props) const override;
    
      // Given a supported NodeDef, returns a XlaLaunchOp that computes the node.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 26 19:43:17 UTC 2023
    - 1.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_kernel_creator.cc

    #include "tensorflow/core/framework/types.h"
    #include "tensorflow/core/lib/core/errors.h"
    #include "tensorflow/core/lib/core/status.h"
    #include "tsl/platform/errors.h"
    
    namespace tensorflow {
    
    bool XlaKernelCreator::CanCreateKernel(
        const FunctionLibraryRuntime& flr,
        const std::shared_ptr<const NodeProperties>& props) const {
      return CanCreateXlaKernel(props->node_def) &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 22:24:01 UTC 2023
    - 4.4K bytes
    - Viewed (0)
Back to top