Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_index_splitter.cc64 int64 index_operand_number = Cast<HloDynamicIndexInstruction>(dynamic_op) in Run()
Dhlo_instructions.h1283 class HloDynamicIndexInstruction : public HloInstruction {
1285 explicit HloDynamicIndexInstruction(HloOpcode opcode, const Shape& shape) in HloDynamicIndexInstruction() function
1305 class HloDynamicSliceInstruction : public HloDynamicIndexInstruction {
1346 class HloDynamicUpdateSliceInstruction : public HloDynamicIndexInstruction {
Dhlo_instructions.cc2236 : HloDynamicIndexInstruction(HloOpcode::kDynamicSlice, shape), in HloDynamicSliceInstruction()
2246 : HloDynamicIndexInstruction(HloOpcode::kDynamicSlice, shape), in HloDynamicSliceInstruction()
2257 : HloDynamicIndexInstruction(HloOpcode::kDynamicUpdateSlice, shape) { in HloDynamicUpdateSliceInstruction()
2266 : HloDynamicIndexInstruction(HloOpcode::kDynamicUpdateSlice, shape) { in HloDynamicUpdateSliceInstruction()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc454 Cast<HloDynamicIndexInstruction>(use)->first_index_operand_number(); in CreateLiteralForConstrainedUses()