Searched refs:resource_inputs (Results 1 – 3 of 3) sorted by relevance
408 resource_inputs = set()417 if input_id in resource_inputs:420 resource_inputs.add(input_id)442 if (op_is_stateful(op) and not resource_inputs
220 std::vector<Output> resource_inputs; member259 std::back_inserter(result->resource_inputs), IncomingEdgeAsOutput); in GetXlaClusterInfo()386 cluster_info.resource_inputs.size()); in GetXlaRunArgs()400 absl::c_copy(cluster_info.resource_inputs, std::back_inserter(xla_run_args)); in GetXlaRunArgs()498 /*resources=*/cluster_info.resource_inputs, in ReplaceNodeWithXlaCompileAndXlaRun()
249 def replace_with_unreplicated_resources(resource_inputs): argument253 for resource in resource_inputs:258 resource_inputs.discard(t)259 resource_inputs.update(to_add)