Searched refs:new_variable_start_index (Results 1 – 1 of 1) sorted by relevance
546 int new_variable_start_index = in MoveHeadOutsideCompilationToHost() local551 xla_node->AddAttr("_variable_start_index", new_variable_start_index); in MoveHeadOutsideCompilationToHost()554 new_variable_start_index + num_variables; in MoveHeadOutsideCompilationToHost()936 int new_variable_start_index = num_new_per_replica_inputs + in RemoveUnusedXlaInput() local941 xla_node->AddAttr("_variable_start_index", new_variable_start_index); in RemoveUnusedXlaInput()944 new_variable_start_index + new_num_variables; in RemoveUnusedXlaInput()