Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.h130 HloInstruction* MakeIotaHlo(HloComputation* computation, const Shape& shape,
Dhlo_creation_utils_test.cc270 HloInstruction* output = MakeIotaHlo(module->entry_computation(), in TEST_F()
Dhlo_creation_utils.cc280 HloInstruction* MakeIotaHlo(HloComputation* computation, const Shape& shape, in MakeIotaHlo() function