Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_test.cc6084 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
6126 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
6168 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
6214 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
6260 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
6283 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
6317 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
6355 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
6403 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
6456 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P()
[all …]
Dhlo_module.h471 CrossProgramPrefetches() const { in CrossProgramPrefetches() function
Dhlo_module.cc321 for (const auto& parameter_indices : CrossProgramPrefetches()) { in ToProto()
900 for (const auto& parameter_indices : CrossProgramPrefetches()) { in Clone()
Dmemory_space_assignment.cc1477 absl::c_count(module->CrossProgramPrefetches(), in Finish()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dtpu_executable_interface.cc209 for (const auto& prefetch : hlo_module_->CrossProgramPrefetches()) { in ExecuteAsyncOnStream()