Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for NodeProperties (0.28 sec)

  1. tensorflow/compiler/jit/xla_kernel_creator.cc

    #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) &&
             !XlaOpRegistry::IsCompilationDevice(flr.device()->device_type());
    }
    
    static Status CreateXlaKernel(FunctionLibraryRuntime* flr,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 22:24:01 UTC 2023
    - 4.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_kernel_creator.h

      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.
      Status CreateKernel(FunctionLibraryRuntime* flr,
                          const std::shared_ptr<const NodeProperties>& props,
                          std::unique_ptr<OpKernel>* kernel) const override;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 26 19:43:17 UTC 2023
    - 1.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_kernel_creator_test.cc

    #include "tensorflow/core/public/version.h"
    
    namespace tensorflow {
    
    std::shared_ptr<NodeProperties> ToNodeProperties(const string& text) {
      NodeDef node_def;
      DataTypeVector dummy;
      EXPECT_TRUE(protobuf::TextFormat::MergeFromString(text, &node_def));
      return std::make_shared<NodeProperties>(nullptr, std::move(node_def), dummy,
                                              dummy);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 16 01:39:55 UTC 2023
    - 5.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc

      std::shared_ptr<const tensorflow::NodeProperties> props;
      absl::Status status = tensorflow::NodeProperties::CreateFromNodeDef(
          *nodedef_or.value(),
          params_.function_library->GetFunctionLibraryDefinition(), &props);
      if (!status.ok()) {
        return op_->emitRemark()
               << "failed to create NodeProperties: " << status.ToString();
      }
      tensorflow::OpKernel* op_kernel_raw;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 18.9K bytes
    - Viewed (0)
Back to top