- Sort Score
- Result 10 results
- Languages All
Results 41 - 44 of 44 for device_num (0.35 sec)
-
tensorflow/compiler/jit/partially_decluster_pass_test.cc
namespace tensorflow { namespace { REGISTER_OP("FakeNullary").Output("out: int32"); REGISTER_OP("FakeBinary") .Input("host_in: int32") .Input("device_in: int32") .Output("host_out: int32") .Output("device_out: int32"); REGISTER_OP("FakeResourceVar").Output("out: resource"); REGISTER_OP("FakeResourceUpdate") .Input("in: resource") .Output("out: resource") .Output("something_else: int32");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 23K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
llvm::SmallVectorImpl<Device>* parsed_devices) { parsed_devices->reserve(device_names.size()); for (const auto& device_name : device_names) { Device parsed_name; if (!DeviceNameUtils::ParseFullName(device_name, &parsed_name)) return false; parsed_devices->push_back(parsed_name); } return true; } using DeviceNames = llvm::SmallVector<std::string, 8>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 46.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
private: std::unique_ptr<MLIRContext> context_; OpBuilder builder_; func::FuncOp func_; OwningOpRef<mlir::ModuleOp> module_; }; Status MlirAbstractOp::Reset(const char* op, const char* device_name) { if (state_) { return FailedPrecondition("Reset called on already built op."); } TF_RETURN_IF_ERROR( tensorflow::OpRegistry::Global()->LookUpOpDef(op, &op_def_)); assert(op_def_);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
for (int i = 0; i < num_devices; ++i) { const char* device_name = TF_DeviceListName(device_list, i, s); CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); const char* device_type = TF_DeviceListType(device_list, i, s); CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); LOG(INFO) << "Device " << i << " has name " << device_name << ", type " << device_type; if (string(device_type) == DEVICE_GPU) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 96.9K bytes - Viewed (0)