Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h277 void set_entry_computation_parameter(int64 parameter_number, in set_entry_computation_parameter() function
Dbuffer_assignment.cc1136 allocation->set_entry_computation_parameter( in AssignSingleHloBuffer()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc1049 allocations->at(index).set_entry_computation_parameter( in GetMlirAllocationInfo()