Home
last modified time | relevance | path

Searched refs:parsed_device_name (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dtf_to_tfrt.cc141 auto parsed_device_name = in matchAndRewrite() local
143 if (!parsed_device_name) in matchAndRewrite()
166 if ((parsed_device_name->device_type == kDeviceTypeTpu && in matchAndRewrite()
177 op, operands, parsed_device_name->op_handler_name, op_attrs, in matchAndRewrite()
257 auto parsed_device_name = in ConvertToFallbackExecuteOp() local
259 if (parsed_device_name && parsed_device_name->device_type == DEVICE_GPU) { in ConvertToFallbackExecuteOp()
634 if (auto parsed_device_name = corert_converter_.ParseDeviceName(op)) in matchAndRewrite() local
635 if (parsed_device_name->device_type != DEVICE_CPU) return failure(); in matchAndRewrite()
815 auto parsed_device_name = corert_converter_.ParseDeviceName(op); in matchAndRewrite() local
817 if (!parsed_device_name) { in matchAndRewrite()
[all …]
Dcorert_converter.cc154 auto parsed_device_name = in ParseDeviceName() local
156 if (!parsed_device_name) op->emitWarning("failed to parse device name."); in ParseDeviceName()
157 return parsed_device_name; in ParseDeviceName()