Searched refs:resource_arg_unique_id (Results 1 – 8 of 8) sorted by relevance
62 resource_arg_unique_id {66 resource_arg_unique_id {83 # Check that the `resource_arg_unique_id` for each argument is propagated to the84 # `tf.resource_arg_unique_id` argument attribute of the function95 # CHECK-SAME: tf.resource_arg_unique_id = 096 # CHECK-SAME tf.resource_arg_unique_id = 0
11 …g0: tensor<*x!tf.resource> {tf.resource_arg_unique_id = 0 : i64}, %arg1: tensor<*x!tf.resource> {t…18 // Check that the `tf.resource_arg_unique_id` argument attributes of20 // resource_arg_unique_id.55 // Check resource_arg_unique_id for each argument. Since they alias each other,57 // CHECK: resource_arg_unique_id60 // CHECK: resource_arg_unique_id
436 int64 resource_arg_unique_id = -1; in GraphToFunctionDef() local452 resource_arg_unique_id = attr.second.i(); in GraphToFunctionDef()458 if (resource_arg_unique_id >= 0) { in GraphToFunctionDef()459 (*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()
49 map<uint32, uint32> resource_arg_unique_id = 8; field
789 %arg0: tensor<*x!tf.resource<tensor<32xf32>>> {tf.resource_arg_unique_id = 0 : i64},790 %arg1: tensor<*x!tf.resource<tensor<32xf32>>> {tf.resource_arg_unique_id = 0 : i64},791 %arg2: tensor<*x!tf.resource<tensor<32xf32>>> {tf.resource_arg_unique_id = 33 : i64}) {
1043 for (const auto& entry : fbody.fdef.resource_arg_unique_id()) { in GetArgsAndRetsFromFunctionBody()2050 tensorflow::int64 resource_arg_unique_id; in GetArgsRetsAndTypesFromFunctionGraph() local2052 &resource_arg_unique_id)) { in GetArgsRetsAndTypesFromFunctionGraph()2054 resource_arg_unique_id); in GetArgsRetsAndTypesFromFunctionGraph()
104 *(fn_stub->mutable_resource_arg_unique_id()) = fn.resource_arg_unique_id(); in GetFunctionDefLibraryStub()