| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_compile_op_replication_pass.cc | 31 constexpr char kDeviceAttr[] = "device"; variable
36 if (StringAttr device = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in GetHost()
73 new_compile_op->setAttr(kDeviceAttr, in runOnOperation()
79 new_assert_op->setAttr(kDeviceAttr, in runOnOperation()
80 new_compile_op->getAttr(kDeviceAttr)); in runOnOperation()
|
| D | constant_op_device_assignment.cc | 32 constexpr const char *kDeviceAttr = "device"; variable
45 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in runOnOperation()
55 StringAttr device_attr = user_op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation()
65 new_op->setAttr(kDeviceAttr, device_attr); in runOnOperation()
|
| D | cluster_ops_by_policy_pass.cc | 45 constexpr char kDeviceAttr[] = "device"; variable
80 auto device = op->getAttr(kDeviceAttr); in IsOplistMatch()
97 if (curr_op->getAttr(kDeviceAttr) != device) return false; in IsOplistMatch()
146 if (auto device = lastOp->getAttr(kDeviceAttr)) in ClusterMatchedOps()
147 clusterOp->setAttr(kDeviceAttr, device); in ClusterMatchedOps()
|
| D | launch_to_device_attribute.cc | 58 constexpr char kDeviceAttr[] = "device"; variable
72 auto device_attr = op->getAttr(kDeviceAttr); in AssignDevicesInRegion()
74 op->setAttr(kDeviceAttr, launch.deviceAttr()); in AssignDevicesInRegion()
80 op->setAttr(kDeviceAttr, launch.deviceAttr()); in AssignDevicesInRegion()
|
| D | tensor_device_copy_conversion.cc | 35 constexpr const char *kDeviceAttr = "device"; variable
59 kDeviceAttr)) { in runOnFunction()
63 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnFunction()
|
| D | tpu_device_propagation.cc | 38 constexpr char kDeviceAttr[] = "device"; variable
63 auto lhs_device_attr = lhs->getAttrOfType<StringAttr>(kDeviceAttr); in IsSupportedExecutorOp()
64 auto rhs_device_attr = rhs->getAttrOfType<StringAttr>(kDeviceAttr); in IsSupportedExecutorOp()
175 auto device_attr = op_to_update->getAttrOfType<StringAttr>(kDeviceAttr); in PropagateDevicesInGraph()
194 op_to_update->setAttr(kDeviceAttr, new_device_attr); in PropagateDevicesInGraph()
201 source->setAttr(kDeviceAttr, new_device_attr); in PropagateDevicesInGraph()
|
| D | tpu_cluster_cleanup_attributes.cc | 31 constexpr char kDeviceAttr[] = "device"; variable
41 for (StringRef attr : {kTPUReplicateAttr, kDeviceAttr}) in runOnOperation()
|
| D | tf_device_replication_pass.cc | 30 constexpr char kDeviceAttr[] = "device"; variable
76 new_op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation()
83 new_op->setAttr(kDeviceAttr, new_devices_array[i].cast<StringAttr>()); in runOnOperation()
|
| D | outside_compiled_to_host_launch.cc | 30 constexpr char kDeviceAttr[] = "device"; variable
77 host_op->removeAttr(Identifier::get(kDeviceAttr, context)); in WrapOpInLaunch()
|
| D | tpu_dynamic_layout_pass.cc | 51 constexpr char kDeviceAttr[] = "device"; variable
123 device = alias.getDefiningOp()->getAttrOfType<StringAttr>(kDeviceAttr); in IsSupportedInputOp()
176 copy_with_layout->setAttr(kDeviceAttr, execute_launch.deviceAttr()); in HandleInput()
210 copy_with_layout->setAttr(kDeviceAttr, in HandleReplicatedInputs()
|
| D | replicate_invariant_op_hoisting.cc | 37 constexpr char kDeviceAttr[] = "device"; variable
122 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in UsesVirtualDevice()
|
| D | resource_device_inference.cc | 53 constexpr char kDeviceAttr[] = "device"; variable
137 auto device_attr = op->getAttrOfType<mlir::StringAttr>(kDeviceAttr); in GetDeviceAttr()
195 identity->setAttr(kDeviceAttr, builder.getStringAttr(*device)); in ComputeResourceDevicesInComputation()
|
| D | cluster_tf_ops_pass.cc | 46 constexpr const char *kDeviceAttr = "device"; variable
67 if (StringAttr attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in GetHost()
149 defining_op->getAttrOfType<StringAttr>(kDeviceAttr)) { in GetFunctionMetadatas()
|
| D | replicate_to_island.cc | 48 constexpr char kDeviceAttr[] = "device"; variable
125 kDeviceAttr, in UpdateRegionReplicateVariantOps()
|
| D | tpu_variable_runtime_reformatting.cc | 60 constexpr char kDeviceAttr[] = "device"; variable
354 builder->getNamedAttr(kDeviceAttr, builder->getStringAttr(device)), in CreateStateVars()
|
| D | tpu_extract_outside_compilation.cc | 50 constexpr char kDeviceAttr[] = "device"; variable
306 op.removeAttr(Identifier::get(kDeviceAttr, op.getContext())); in MoveOpsToHost()
|
| D | tpu_cluster_formation.cc | 53 constexpr char kDeviceAttr[] = "device"; variable
276 inner_op->removeAttr(kDeviceAttr); in CreateClusterOp()
|
| D | tpu_merge_variables_with_execute.cc | 55 constexpr char kDeviceAttr[] = "device"; variable
159 auto resource_attr = resource_op->getAttr(kDeviceAttr); in BuildVariableAccessInfo()
|
| D | tpu_space_to_depth_pass.cc | 54 constexpr char kDeviceAttr[] = "device"; variable
412 auto device = op->getAttrOfType<StringAttr>(kDeviceAttr); in IsSupportedHostInputOp()
|
| D | tpu_rewrite_pass.cc | 70 constexpr char kDeviceAttr[] = "device"; variable
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 80 constexpr char kDeviceAttr[] = "tf.device"; variable
239 func.getArgAttrOfType<mlir::StringAttr>(index, kDeviceAttr)) in GetArgumentNode()
244 absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr}; in GetArgumentNode()
274 function.getResultAttrOfType<mlir::StringAttr>(index, kDeviceAttr)) in GetReturnNode()
279 absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr}; in GetReturnNode()
690 kDeviceAttr, kResourceArgUniqueIdAttr}; in ConvertLibFunction()
|