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.h1576 class HloDynamicIndexInstruction : public HloInstruction {
1578 explicit HloDynamicIndexInstruction(HloOpcode opcode, const Shape& shape) in HloDynamicIndexInstruction() function
1598 class HloDynamicSliceInstruction : public HloDynamicIndexInstruction {
1639 class HloDynamicUpdateSliceInstruction : public HloDynamicIndexInstruction {
Dhlo_instructions.cc2649 : HloDynamicIndexInstruction(HloOpcode::kDynamicSlice, shape), in HloDynamicSliceInstruction()
2659 : HloDynamicIndexInstruction(HloOpcode::kDynamicSlice, shape), in HloDynamicSliceInstruction()
2670 : HloDynamicIndexInstruction(HloOpcode::kDynamicUpdateSlice, shape) { in HloDynamicUpdateSliceInstruction()
2679 : HloDynamicIndexInstruction(HloOpcode::kDynamicUpdateSlice, shape) { in HloDynamicUpdateSliceInstruction()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc565 Cast<HloDynamicIndexInstruction>(use)->first_index_operand_number(); in CreateLiteralForConstrainedUses()