- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for getFuncAttr (0.1 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
has_resources = true; } } if (!in_order) { // Functions do not get reused in practice, so skip the check for if the // callee has been updated. StringAttr callee_sym = cluster_func_op.getFuncAttr().getAttr(); MoveResourceArgsToEnd(symtab.lookup<func::FuncOp>(callee_sym)); } builder.setInsertionPoint(cluster_func_op); auto xla_launch_op = builder.create<TF::XlaLaunchOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc
} } } cluster_func_op->setAttr(TF::kDynamicArgIndexAttr, builder.getI32ArrayAttr(dynamic_shape_arg_index)); FlatSymbolRefAttr func_attr = cluster_func_op.getFuncAttr(); func::FuncOp func = cluster_func_op->getParentOfType<ModuleOp>().lookupSymbol<func::FuncOp>( func_attr.getValue()); // Update the marked argument with dynamic shapes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_input_output_aliases.cc
if (failed(BuildAliasingInfo(cluster_func, resource_alias_info_map)) || resource_alias_info_map.empty()) { return; } FlatSymbolRefAttr func_attr = cluster_func.getFuncAttr(); func::FuncOp device_func = module.lookupSymbol<func::FuncOp>(func_attr.getValue()); AddAliasingAttributeToDeviceFunc(device_func, resource_alias_info_map); }); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 04:14:26 UTC 2024 - 7.5K bytes - Viewed (0)