- Sort Score
- Result 10 results
- Languages All
Results 11 - 15 of 15 for DeviceSet (0.16 sec)
-
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass_test.cc
std::vector<std::unique_ptr<Device>> devices; devices.push_back(FakeDevice::Make(kDeviceName, DEVICE_GPU)); devices.push_back(FakeDevice::Make(kHostName, DEVICE_CPU)); std::unique_ptr<DeviceSet> device_set(new DeviceSet()); for (auto& device : devices) { device_set->AddDevice(device.get()); } auto graph = std::make_unique<Graph>(OpRegistry::Global()); SessionOptions session_options;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
#include "tensorflow/core/public/version.h" #include "tensorflow/core/util/dump_graph.h" namespace tensorflow { namespace { using DeadnessPredicate = DeadnessAnalysis::DeadnessPredicate; using jit::DeviceId; using jit::DeviceSet; // The clusters we create here are eventually lowered into an // _XlaCompile/_XlaRun pair with a TF executor "fallback" that uses the // PartitionedCall op to execute the cluster in the regular graph executor if
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc
mlir::func::registerAllExtensions(registry); // clang-format on } Status MlirFunctionOptimizationPass::Run( const std::string& function_name, const DeviceSet& device_set, const ConfigProto& config_proto, const FunctionOptimizationPass::FunctionOptions& function_options, std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def,
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/c/eager/c_api_distributed_test.cc
: error_node_(error_node), error_device_(error_device) {} tensorflow::Status Run(const std::string& function_name, const tensorflow::DeviceSet& device_set, const tensorflow::ConfigProto& config_proto, const FunctionOptions& function_options, std::unique_ptr<tensorflow::Graph>* graph,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass.cc
const string& function_name, const FunctionLibraryDefinition& flib_def) { const FunctionDef* func_def = flib_def.Find(function_name); TF_RET_CHECK(func_def) << "Could not find " << function_name; jit::DeviceSet device_set; for (const NodeDef& ndef : func_def->node_def()) { VLOG(3) << ndef.DebugString(); if (!ndef.device().empty()) { TF_ASSIGN_OR_RETURN(jit::DeviceId device_id,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 24.3K bytes - Viewed (0)