Home
last modified time | relevance | path

Searched refs:GetDeviceOrdinalFromDeviceString (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Ddevice_util_test.cc210 TEST(DeviceUtilTest, GetDeviceOrdinalFromDeviceString) { in TEST() argument
219 GetDeviceOrdinalFromDeviceString(unknown_loc, tpu0, &device_ordinal0); in TEST()
225 GetDeviceOrdinalFromDeviceString(unknown_loc, tpu1, &device_ordinal1); in TEST()
235 mlir::LogicalResult result = GetDeviceOrdinalFromDeviceString( in TEST()
248 GetDeviceOrdinalFromDeviceString(unknown_loc, tpu_no_id, &device_ordinal); in TEST()
H A Ddevice_util.h47 mlir::LogicalResult GetDeviceOrdinalFromDeviceString(mlir::Location loc,
H A Ddevice_util.cc158 mlir::LogicalResult GetDeviceOrdinalFromDeviceString(mlir::Location loc, in GetDeviceOrdinalFromDeviceString() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dreplicate_to_island.cc78 if (succeeded(tensorflow::GetDeviceOrdinalFromDeviceString( in GetDeviceOrdinal()