Home
last modified time | relevance | path

Searched refs:AddCrossProgramPrefetch (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module.cc484 module->AddCrossProgramPrefetch( in CreateFromProto()
903 module->AddCrossProgramPrefetch(parameter, indices); in Clone()
H A Dhlo_module.h465 void AddCrossProgramPrefetch(int64_t parameter, const ShapeIndex& index) { in AddCrossProgramPrefetch() function
H A Dmemory_space_assignment.cc2117 module->AddCrossProgramPrefetch(parameter, buffer->index()); in AllocateCrossProgramPrefetchBuffer()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_execute.cc477 module->AddCrossProgramPrefetch( in TPUExecute()