Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbuffer_assignment.h276 void set_entry_computation_parameter(int64_t parameter_number, in set_entry_computation_parameter() function
H A Dbuffer_assignment.cc1192 allocation->set_entry_computation_parameter( in AssignSingleHloBuffer()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc911 .set_entry_computation_parameter( in SetUpMlirAllocation()