Searched refs:entry_parameter_number (Results 1 – 3 of 3) sorted by relevance
47 if (buffer_infos[i].entry_parameter_number() >= result.size()) { in CreateArgIndexTableFromBufferInfos()48 result.resize(buffer_infos[i].entry_parameter_number() + 1); in CreateArgIndexTableFromBufferInfos()50 result[buffer_infos[i].entry_parameter_number()] = i; in CreateArgIndexTableFromBufferInfos()
50 uint64 entry_parameter_number() const { in entry_parameter_number() function82 entry_parameter_number() == buffer_info.entry_parameter_number();
1702 int64 entry_parameter_number = -1; in AssignColocatedBufferSets() local1710 entry_parameter_number = instruction->parameter_number(); in AssignColocatedBufferSets()1717 CHECK(!is_constant || entry_parameter_number == -1) in AssignColocatedBufferSets()1743 if (entry_parameter_number >= 0) { in AssignColocatedBufferSets()1746 entry_parameter_number, *entry_parameter_shape_idx); in AssignColocatedBufferSets()1747 allocation->set_entry_computation_parameter(entry_parameter_number, in AssignColocatedBufferSets()