Home
last modified time | relevance | path

Searched refs:CrossProgramPrefetches (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable_interface.cc194 for (const auto& prefetch : hlo_module_->CrossProgramPrefetches()) { in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module.h367 const absl::Span<const std::pair<int64, ShapeIndex>> CrossProgramPrefetches() in CrossProgramPrefetches() function
Dhlo_module.cc289 for (const auto& parameter_indices : CrossProgramPrefetches()) { in ToProto()
731 for (const auto& parameter_indices : CrossProgramPrefetches()) { in Clone()
Dmemory_space_assignment_test.cc4853 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
4899 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
4945 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
4968 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
5006 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
5059 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
5100 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
5132 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
5192 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
Dmemory_space_assignment.cc1141 absl::c_count(module->CrossProgramPrefetches(), in Finish()