Searched refs:StreamExecutorConfig (Results 1 – 16 of 16) sorted by relevance
49 StreamExecutorConfig config; in InspectNumaNodes()74 StreamExecutorConfig config; in DeviceToBus()86 StreamExecutorConfig config; in FirstExecutorForBus()113 StreamExecutorConfig config; in ExecutorForDevice()122 StreamExecutorConfig config; in ExecutorForDeviceWithPluginConfig()130 const StreamExecutorConfig& config) { in GetExecutor()136 ROCmPlatform::GetUncachedExecutor(const StreamExecutorConfig& config) { in GetUncachedExecutor()
73 const StreamExecutorConfig& config) override;76 const StreamExecutorConfig& config) override;
84 StreamExecutorConfig config; in InspectNumaNodes()109 StreamExecutorConfig config; in DeviceToBus()121 StreamExecutorConfig config; in FirstExecutorForBus()147 StreamExecutorConfig config; in ExecutorForDevice()156 StreamExecutorConfig config; in ExecutorForDeviceWithPluginConfig()164 const StreamExecutorConfig& config) { in GetExecutor()170 CudaPlatform::GetUncachedExecutor(const StreamExecutorConfig& config) { in GetUncachedExecutor()
75 const StreamExecutorConfig& config) override;78 const StreamExecutorConfig& config) override;
72 struct StreamExecutorConfig { struct75 StreamExecutorConfig();78 explicit StreamExecutorConfig(int ordinal);154 const StreamExecutorConfig& config) = 0;160 const StreamExecutorConfig& config) = 0;
40 const StreamExecutorConfig& config,45 port::StatusOr<StreamExecutor*> Get(const StreamExecutorConfig& config);65 std::pair<StreamExecutorConfig, std::unique_ptr<StreamExecutor>>>
83 StreamExecutorConfig::StreamExecutorConfig() in StreamExecutorConfig() function in stream_executor::StreamExecutorConfig86 StreamExecutorConfig::StreamExecutorConfig(int ordinal_in) in StreamExecutorConfig() function in stream_executor::StreamExecutorConfig
23 const StreamExecutorConfig& config, in GetOrCreate()66 const StreamExecutorConfig& config) { in Get()
28 StreamExecutorConfig config(/*ordinal=*/0); in NewStreamExecutor()
45 StreamExecutorConfig config; in ExecutorForDevice()54 StreamExecutorConfig config; in ExecutorForDeviceWithPluginConfig()62 const StreamExecutorConfig& config) { in GetExecutor()68 HostPlatform::GetUncachedExecutor(const StreamExecutorConfig& config) { in GetUncachedExecutor()
61 const StreamExecutorConfig& config) override;64 const StreamExecutorConfig& config) override;
48 StreamExecutorConfig config; in ExecutorForDevice()58 StreamExecutorConfig config; in ExecutorForDeviceWithPluginConfig()66 const StreamExecutorConfig& config) { in GetExecutor()73 const StreamExecutorConfig& config) { in GetUncachedExecutor()
48 const StreamExecutorConfig& config) override;51 const StreamExecutorConfig& config) override;
33 platform->GetExecutor(se::StreamExecutorConfig(/*ordinal=*/0)) in TEST()52 platform->GetExecutor(se::StreamExecutorConfig(/*ordinal=*/0)) in TEST()86 platform->GetExecutor(se::StreamExecutorConfig(/*ordinal=*/0)) in TEST()146 platform->GetExecutor(se::StreamExecutorConfig(/*ordinal=*/0)) in TEST()251 platform->GetExecutor(se::StreamExecutorConfig(/*ordinal=*/0)) in TEST()
31 se::StreamExecutorConfig config(/*ordinal=*/0); in NewStreamExecutor()
247 se::StreamExecutorConfig config; in GetStreamExecutors()