- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 45 for getDevice (0.16 sec)
-
tensorflow/compiler/jit/xla_platform_info_test.cc
}; #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM TEST_F(XlaPlatformInfoTest, BuildXlaDeviceCompilerXlaDeviceMetadata) { device_setup_.AddDevicesAndSetUp({DEVICE_XLA_GPU}); Device* device = device_setup_.GetDevice(DEVICE_XLA_GPU); const XlaDevice::Metadata* metadata = nullptr; TF_CHECK_OK(XlaDevice::GetMetadataFromDevice(device, &metadata)); XlaPlatformInfo platform_info = XlaPlatformInfoFromDevice(device);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Jan 14 15:17:12 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util_test.cc
} DeviceSetup device_setup_; }; TEST_F(XlaCompilerOptionsTest, PjRtOptionsXlaDevice) { device_setup_.AddDevicesAndSetUp({DEVICE_XLA_GPU}); Device* device = device_setup_.GetDevice(DEVICE_XLA_GPU); DeviceType compilation_device_type = DeviceType(DEVICE_GPU_XLA_JIT); se::Platform::Id platform_id = nullptr; auto xla_device_metadata = CreateXlaDeviceMetadata(compilation_device_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 29 01:41:20 UTC 2023 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
rewriter.replaceOp(op_to_replace, ValueRange({fused_op})); return success(); } }; const char kDeviceAttr[] = "device"; const char kDeviceGpu[] = "GPU"; std::optional<std::string> GetDevice(mlir::Operation *op) { mlir::StringAttr device = op->getAttrOfType<mlir::StringAttr>(kDeviceAttr); if (!device || device.getValue().empty()) { return std::nullopt; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
execute_launch.getLoc(), llvm::ArrayRef<Type>{}, reformat_operands); WrapOpInLaunch(&builder, execute_launch.getLoc(), reformat_op, execute_launch.getDevice()); // Build the replicated unformat op after the loop. First prepare building the // replicate op. llvm::SmallVector<std::pair<ValueRange, Type>, 8> unformat_replicate_operands;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
BuildCopyWithLayout(execute_launch, compile_launch, get_layout, entry.value().get(), &builder); auto device_list = mlir::cast<ArrayAttr>( replicate.getDevices().value().get(execute_launch.getDevice())); copy_with_layout->setAttr(kDeviceAttr, device_list.getValue()[entry.index()]); entry.value().set(copy_with_layout); } return true; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc
// Map aliased devices to explicit devices based on replica. if (auto launch = dyn_cast<tf_device::LaunchOp>(op)) if (auto device_by_replica = devices.value().get(launch.getDevice())) launch->setAttr( kDeviceAttr, device_by_replica.cast<ArrayAttr>()[replica_id].cast<StringAttr>()); return WalkResult::advance(); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0) -
maven-api-impl/src/main/java/org/apache/maven/internal/impl/AbstractSession.java
// /** * Shortcut for <code>getService(RepositoryFactory.class).createLocal(...)</code> * * @see RepositoryFactory#createLocal(Path) */ @Override public LocalRepository createLocalRepository(Path path) { return getService(RepositoryFactory.class).createLocal(path); } /** * Shortcut for <code>getService(RepositoryFactory.class).createRemote(...)</code> *
Registered: Wed Jun 12 09:55:16 UTC 2024 - Last Modified: Tue Jun 11 07:23:04 UTC 2024 - 27.5K bytes - Viewed (0) -
api/maven-api-core/src/main/java/org/apache/maven/api/Session.java
* <p> * Shortcut for {@code getService(TypeRegistry.class).require(...)}. * * @see org.apache.maven.api.services.TypeRegistry#require(String) */ @Nonnull Type requireType(@Nonnull String id); /** * Obtain the {@link Language} from the specified {@code id}. * <p> * Shortcut for {@code getService(LanguageRegistry.class).require(...)}. *
Registered: Wed Jun 12 09:55:16 UTC 2024 - Last Modified: Tue Jun 11 07:23:04 UTC 2024 - 30.2K bytes - Viewed (0) -
pkg/kubelet/apis/podresources/testing/provider_mock.go
} // GetDevices mocks base method. func (m *MockDevicesProvider) GetDevices(podUID, containerName string) []*v10.ContainerDevices { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetDevices", podUID, containerName) ret0, _ := ret[0].([]*v10.ContainerDevices) return ret0 } // GetDevices indicates an expected call of GetDevices.
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Thu Mar 07 08:12:16 UTC 2024 - 10.1K bytes - Viewed (0) -
pilot/pkg/serviceregistry/kube/controller/serviceimportcache_test.go
} // Wait for the resources to be processed by the controller. retry.UntilSuccessOrFail(t, func() error { if svc := ic.GetService(ic.clusterLocalHost()); svc != nil { return fmt.Errorf("found deleted service for host %s", ic.clusterLocalHost()) } if svc := ic.GetService(serviceImportClusterSetHost); svc != nil { return fmt.Errorf("found deleted service for host %s", serviceImportClusterSetHost) }
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Mon Aug 14 18:50:38 UTC 2023 - 15.2K bytes - Viewed (0)