- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for NodeProperties (0.27 sec)
-
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) -
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) -
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)