Home
last modified time | relevance | path

Searched refs:entry_parameter_number (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dbuffer_info_util.cc47 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()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dcpu_function_runtime.h50 uint64 entry_parameter_number() const { in entry_parameter_number() function
82 entry_parameter_number() == buffer_info.entry_parameter_number();
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc1702 int64 entry_parameter_number = -1; in AssignColocatedBufferSets() local
1710 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()