Searched refs:new_variable_start_index (Results 1 – 1 of 1) sorted by relevance
574 int new_variable_start_index = in MoveHeadOutsideCompilationToHost() local579 xla_node->AddAttr("_variable_start_index", new_variable_start_index); in MoveHeadOutsideCompilationToHost()582 new_variable_start_index + num_variables; in MoveHeadOutsideCompilationToHost()967 int new_variable_start_index = num_new_per_replica_inputs + in RemoveUnusedXlaInput() local972 xla_node->AddAttr("_variable_start_index", new_variable_start_index); in RemoveUnusedXlaInput()975 new_variable_start_index + new_num_variables; in RemoveUnusedXlaInput()