| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | resource_device_inference.cc | 132 auto device_attr = in GetDeviceAttr() local
139 auto device_attr = op->getAttrOfType<mlir::StringAttr>(kDeviceAttr); in GetDeviceAttr() local
156 StringRef device_attr = GetDeviceAttr(func_op, arg.getArgNumber()); in ComputeResourceDevicesInComputation() local
177 StringRef device_attr = GetDeviceAttr(op); in ComputeResourceDevicesInComputation() local
|
| H A D | constant_op_device_assignment.cc | 46 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in runOnOperation() local
56 StringAttr device_attr = user_op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation() local
|
| H A D | tf_device_assignment.cc | 43 if (auto device_attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in runOnOperation() local
80 if (auto device_attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in runOnOperation() local
|
| H A D | tpu_device_propagation.cc | 178 auto device_attr = op_to_update->getAttrOfType<StringAttr>(kDeviceAttr); in PropagateDevicesInGraph() local
221 auto device_attr = func.getResultAttrOfType<StringAttr>( in PropagateDevicesToResults() local
|
| H A D | host_launch_to_outside_compiled.cc | 73 StringAttr device_attr = launch->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation() local
|
| H A D | launch_to_device_attribute.cc | 45 auto device_attr = op->getAttr(kDeviceAttr); in AssignDevicesInRegion() local
|
| H A D | readonly_references_to_resources.cc | 153 StringAttr device_attr = in runOnOperation() local
|
| H A D | cluster_formation.cc | 56 auto device_attr = op->getAttrOfType<StringAttr>("device"); in GetDevice() local
|
| H A D | tpu_merge_variables_with_execute.cc | 156 Attribute device_attr = execute_launch.deviceAttr(); in BuildVariableAccessInfo() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | cross_device_transfer.cc | 78 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in GetDevice() local
98 if (StringAttr device_attr = parent_func_op.getArgAttrOfType<StringAttr>( in GetDevice() local
|
| H A D | lower_saved_model.cc | 125 if (auto device_attr = in ReplaceHoistedValues() local
385 if (auto device_attr = new_op->getAttrOfType<mlir::StringAttr>("device")) { in HoistInvariantOps() local
651 if (auto device_attr = op->getAttrOfType<mlir::StringAttr>("device")) { in runOnOperation() local
|
| H A D | fallback_converter.cc | 90 if (auto device_attr = def->getAttrOfType<mlir::StringAttr>("device")) { in ConvertFallbackTensorToCoreRTTensorHandle() local
|
| H A D | corert_converter.cc | 149 auto device_attr = op->getAttr("device"); in ParseDeviceName() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| H A D | c_api_multi_client_test.cc | 131 for (const auto& device_attr : device_attrs) { in TEST() local
178 for (const auto& device_attr : device_attrs) { in TEST() local
|
| H A D | c_api_multi_client_function_test.cc | 243 for (const auto& device_attr : device_attrs) { in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_session.cc | 395 for (const auto& device_attr : resp.local_device()) { in ListDevices() local
398 for (const auto& device_attr : resp.remote_device()) { in ListDevices() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | export_graphdef.cc | 218 if (auto device_attr = in GetArgumentNode() local
253 if (auto device_attr = in GetReturnNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_op_interfaces.cc | 22 if (auto device_attr = op->getAttrOfType<StringAttr>("device")) in GetDeviceOrEmpty() local
|
| H A D | tf_ops_device_helper.cc | 54 auto device_attr = op->getAttrOfType<StringAttr>("device"); in CanUseGpuDevice() local
|
| H A D | tf_device.cc | 482 for (auto device_attr : op.devices().getValue().getValue()) { in verify() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/ |
| H A D | tf_op_wrapper.h | 132 StringAttr device_attr = deviceAttr(); in device() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| H A D | get_alternative_subgraph.cc | 290 auto device_attr = func->getAttrOfType<StringAttr>(kDevice); in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
| H A D | functiondef_import.cc | 162 StringAttr device_attr = tfgDialect->getDeviceAttrIdentifier(); in ImportNodes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | runtime_fallback_kernels.cc | 1078 RemainingArguments args, RemainingResults results, StringAttr device_attr, in RuntimeFallbackExecuteOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| H A D | pass.cc | 176 if (StringAttr device_attr = switch_op.deviceAttr()) in GetControlAnchorForSwitchResult() local
732 StringAttr device_attr = TFOp(op).deviceAttr(); in matchAndRewrite() local
|