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