- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for UsePjRtForSingleDeviceCompilation (0.43 sec)
-
tensorflow/compiler/jit/xla_compile_util_test.cc
} TEST(XlaCompileUtilTest, PjRtXlaLaunchFlagTest) { EXPECT_FALSE(UsePjRtForSingleDeviceCompilation(DeviceType(DEVICE_CPU))); // Flag is turned on, but no device is allowlisted. auto& rollout_config = GetXlaOpsCommonFlags()->tf_xla_use_device_api; rollout_config.enabled_for_xla_launch_ = true; EXPECT_FALSE(UsePjRtForSingleDeviceCompilation(DeviceType(DEVICE_CPU)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 21:48:05 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_util.h
absl::Span<const DataType> result_types); // Checks if single device compilation and execution with PJRT is enabled for // `device_type` in either the XlaLaunch op or the XlaCompileOnDemand op. bool UsePjRtForSingleDeviceCompilation(const DeviceType& device_type); // Gets the resource name of the PjRt DeviceCompiler for `device_type`. std::string GetPjRtDeviceCompilerResourceName(const DeviceType& device_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_util.cc
.Attr("index", i) .Finalize(graph.get(), &node); TF_RETURN_IF_ERROR(status); } FixupSourceAndSinkEdges(graph.get()); return graph; } bool UsePjRtForSingleDeviceCompilation(const DeviceType& device_type) { const auto& rollout_config = GetXlaOpsCommonFlags()->tf_xla_use_device_api; return rollout_config.IsEnabledInXlaLaunchForDevice(device_type) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 4.6K bytes - Viewed (0)