Searched refs:resource_arg_unique_id (Results 1 – 6 of 6) sorted by relevance
20 // resource_arg_unique_id.54 // Check resource_arg_unique_id for each argument. Since they alias each other,56 // CHECK: resource_arg_unique_id59 // CHECK: resource_arg_unique_id
62 resource_arg_unique_id {66 resource_arg_unique_id {83 # Check that the `resource_arg_unique_id` for each argument is propagated to the
430 int64 resource_arg_unique_id = -1; in GraphToFunctionDef() local456 resource_arg_unique_id = attr.second.i(); in GraphToFunctionDef()462 if (resource_arg_unique_id >= 0) { in GraphToFunctionDef()463 (*fdef->mutable_resource_arg_unique_id())[idx] = resource_arg_unique_id; in GraphToFunctionDef()
174 bool ints_on_device, int64 resource_arg_unique_id) { in BuildInputArgIndex() argument201 if (resource_arg_unique_id >= 0) { in BuildInputArgIndex()202 AddAttr("_resource_arg_unique_id", resource_arg_unique_id, gnode); in BuildInputArgIndex()737 auto resource_id_it = fdef.resource_arg_unique_id().find(i); in InstantiateFunction()738 int64 resource_arg_unique_id = in InstantiateFunction() local739 resource_id_it != fdef.resource_arg_unique_id().end() in InstantiateFunction()743 ints_on_device, resource_arg_unique_id); in InstantiateFunction()
51 map<uint32, uint32> resource_arg_unique_id = 8; field
108 *(fn_stub->mutable_resource_arg_unique_id()) = fn.resource_arg_unique_id(); in GetFunctionDefLibraryStub()