Searched refs:resource_arg (Results 1 – 4 of 4) sorted by relevance
159 auto resource_arg = in LiftVariables() local161 if (!resource_arg) continue; in LiftVariables()163 StringRef resource_name = resource_arg.getValue(); in LiftVariables()
123 BlockArgument resource_arg, in ValidateResourceArgument() argument128 << resource_arg.getArgNumber() << " to have one subtype, got " in ValidateResourceArgument()131 auto composite_ops = GetCompositeResourceUserNames(resource_arg); in ValidateResourceArgument()135 << resource_arg.getArgNumber() in ValidateResourceArgument()
163 auto resource_arg = resource.dyn_cast<BlockArgument>(); in BuildVariableAccessInfo() local164 assert(resource_arg); in BuildVariableAccessInfo()165 if (resource_arg.getOwner() != &func.front()) continue; in BuildVariableAccessInfo()168 resource_arg.getArgNumber(), kFuncDeviceAttr); in BuildVariableAccessInfo()
572 // CHECK-LABEL: func @resource_arg573 func @resource_arg(%arg0: tensor<*x!tf.resource>) -> tensor<*x!tf.resource> {