- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 99 for registry_ (0.15 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/logging_hooks_test.cc
"tensorflow/compiler/mlir/tf2xla/internal/testdata/"); } class LoggingHooksTest : public ::testing::Test { public: LoggingHooksTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); env_ = Env::Default(); test_group_name_ = "TestGroup"; test_dir_ = testing::TmpDir();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/cc/framework/grad_op_registry.cc
return grad_op_registry; } bool GradOpRegistry::Register(const string& op, GradFunc func) { CHECK(registry_.insert({op, func}).second) << "Existing gradient for " << op; return true; } Status GradOpRegistry::Lookup(const string& op, GradFunc* func) const { auto iter = registry_.find(op); if (iter == registry_.end()) { const string error_msg = "No gradient defined for op: " + op + ". Please see "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 1.6K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad_test.cc
} AbstractContextPtr immediate_execution_ctx_; GradientRegistry registry_; Status status_; public: bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; } bool UseFunction() const { return std::get<2>(GetParam()); } }; TEST_P(CppGradients, TestReluGrad) { status_ = registry_.Register("Relu", ReluRegisterer); ASSERT_EQ(errors::OK, status_.code()) << status_.message();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/cluster_tf_test.cc
} class SessionClusterTensorflowDialectTest : public ::testing::Test { public: SessionClusterTensorflowDialectTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); } absl::Status CreateMlirModule(std::string mlir_module_filename) { std::string mlir_module_path = TestDataPath() + mlir_module_filename;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/tf_dialect_to_executor_test.cc
} class TensorflowDialectToExecutorTest : public ::testing::Test { public: TensorflowDialectToExecutorTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); } absl::Status CreateMlirModule(std::string mlir_module_filename) { std::string mlir_module_path = TestDataPath() + mlir_module_filename;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_dialect_to_executor_test.cc
} return count; } class TensorflowDialectToExecutorTest : public ::testing::Test { public: TensorflowDialectToExecutorTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); } absl::Status CreateMlirModule(std::string mlir_module_filename) { std::string mlir_module_path = TestDataPath() + mlir_module_filename;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 23:22:50 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf_test.cc
class FunctionClusterTensorflowDialectTest : public ::testing::Test { public: FunctionClusterTensorflowDialectTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); } absl::Status CreateMlirModule(std::string mlir_module_filename) { std::string mlir_module_path = TestDataPath() + mlir_module_filename;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc
<< ", FallbackEnabled: " << num_passes_fallback_enabled << ", Total: " << registry_->passes().size(); } GraphDebugInfo debug_info; mlir::DialectRegistry registry; RegisterDialects(registry); mlir::MLIRContext context(registry); GraphImportConfig import_config; import_config.graph_as_function = true; import_config.control_outputs = *control_ret_node_names;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 18.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops_test.cc
class LowerClusterToRuntimeOpsTest : public ::testing::Test { public: LowerClusterToRuntimeOpsTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); env_ = Env::Default(); test_group_name_ = "TestGroup"; test_dir_ = testing::TmpDir();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 6.5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/array_grad_test.cc
ASSERT_EQ(errors::OK, status_.code()) << status_.message(); x2.reset(x2_raw); } status_ = registry_.Register("IdentityN", IdentityNRegisterer); ASSERT_EQ(errors::OK, status_.code()) << status_.message(); auto IdentityNGradModel = BuildGradModel(IdentityNModel, registry_); std::vector<AbstractTensorHandle*> outputs(2); status_ = RunModel(IdentityNGradModel, immediate_execution_ctx_.get(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 5K bytes - Viewed (0)