- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for default_container (0.28 sec)
-
tensorflow/compiler/jit/xla_platform_info.cc
// within) and create a new one. TF_RETURN_IF_ERROR(rm->Delete<PjRtDeviceCompiler>(rm->default_container(), compiler_name)); TF_RETURN_IF_ERROR(rm->Delete<DeviceCompilationProfiler>( rm->default_container(), profiler_name)); deleted_old_device_compiler = true; } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 17:23:27 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
Var* var = CreateVariable<T>(name, shape, data); ResourceMgr* rm = device_->resource_manager(); TF_ASSERT_OK(rm->Create(rm->default_container(), name, var)); ResourceHandle handle; handle.set_device(device_->name()); handle.set_container(rm->default_container()); handle.set_name(name); TypeIndex type_index = TypeIndex::Make<Var>(); handle.set_hash_code(type_index.hash_code());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/session_utils.cc
tensorflow::Var* var_ptr = nullptr; const auto& container = var_handle_op.getContainer().str(); auto status = device->resource_manager()->Lookup( (container.empty() ? device->resource_manager()->default_container() : container), var_handle_op.getSharedName().str(), &var_ptr); if (!device || !status.ok()) return nullptr; return var_ptr; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
TF_ASSIGN_OR_RETURN(DeviceType compilation_device_type, GetCompilationDeviceType(platform_info_.device_type())); TF_RETURN_IF_ERROR(rm->LookupOrCreate<XlaDeviceCompiler>( rm->default_container(), "xla_device_compiler", xla_device_compiler, [&](XlaDeviceCompiler** xla_device_compiler) { return BuildXlaDeviceCompiler(ctx->device(), ctx->function_library(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
} absl::Status ResetPjRtClient(const DeviceType& device_type) { ResourceMgr* rmgr = tfrt_global::GetTFGlobalResourceMgr(); PjRtState* pjrt_state; TF_RETURN_IF_ERROR(rmgr->Lookup(rmgr->default_container(), kPjRtStateResourceName, &pjrt_state)); TF_RETURN_IF_ERROR(pjrt_state->MovePjRtClientToUnused(device_type)); return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/fake_session.cc
tensorflow::Device* device = nullptr; auto status = device_mgr_->LookupDevice(kDeviceName, &device); if (status != absl::OkStatus()) return; auto container = device->resource_manager()->default_container(); // Create 2 resources and initialize them with dummy values. TF_CHECK_OK(device->resource_manager()->Create( container, "var1", new tensorflow::Var(tensorflow::DataType::DT_FLOAT)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
Var* var = CreateVariable<T>(name, shape, data); ResourceMgr* rm = device_->resource_manager(); TF_ASSERT_OK(rm->Create(rm->default_container(), name, var)); ResourceHandle handle; handle.set_device(device_->name()); handle.set_container(rm->default_container()); handle.set_name(name); TypeIndex type_index = TypeIndex::Make<Var>(); handle.set_hash_code(type_index.hash_code());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
} XlaDeviceCompiler* xla_device_compiler; TF_RETURN_IF_ERROR(dev->resource_manager()->LookupOrCreate<XlaDeviceCompiler>( dev->resource_manager()->default_container(), "xla_device_compiler", &xla_device_compiler, [&](XlaDeviceCompiler** xla_device_compiler) { return BuildXlaDeviceCompiler(dev, flr, platform_info,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/c/kernels.cc
TF_StringView default_container_name_sv; default_container_name_sv.data = cc_ctx->resource_manager()->default_container().data(); default_container_name_sv.len = cc_ctx->resource_manager()->default_container().length(); return default_container_name_sv; } TF_StringView TF_GetOpKernelRequestedInput(TF_OpKernelContext* ctx,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
GetCompilationDeviceType(platform_info.device_type())); XlaDeviceCompiler* xla_device_compiler; TF_RETURN_IF_ERROR(rm->LookupOrCreate<XlaDeviceCompiler>( rm->default_container(), "xla_device_compiler", &xla_device_compiler, [&](XlaDeviceCompiler** xla_device_compiler) { return BuildXlaDeviceCompiler(ctx->device(), ctx->function_library(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0)