Searched refs:HloDimensionsInstruction (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 40 class HloDimensionsInstruction : public HloInstruction {
63 HloDimensionsInstruction(HloOpcode opcode, const Shape& shape, in HloDimensionsInstruction() function
821 class HloReverseInstruction : public HloDimensionsInstruction {
837 class HloConcatenateInstruction : public HloDimensionsInstruction {
858 class HloReduceInstruction : public HloDimensionsInstruction {
894 class HloSortInstruction : public HloDimensionsInstruction {
929 class HloTransposeInstruction : public HloDimensionsInstruction {
947 class HloBroadcastInstruction : public HloDimensionsInstruction {
|
| D | hlo_instructions.cc | 1096 : HloDimensionsInstruction(HloOpcode::kReverse, shape, dimensions) { in HloReverseInstruction()
1100 HloInstructionProto HloDimensionsInstruction::ToProto() const { in ToProto()
1108 std::vector<std::string> HloDimensionsInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()
1113 bool HloDimensionsInstruction::IdenticalSlowPath( in IdenticalSlowPath()
1118 static_cast<const HloDimensionsInstruction&>(other); in IdenticalSlowPath()
1133 : HloDimensionsInstruction(HloOpcode::kConcatenate, shape, {dimension}) { in HloConcatenateInstruction()
1151 : HloDimensionsInstruction(HloOpcode::kReduce, shape, in HloReduceInstruction()
1182 : HloDimensionsInstruction(HloOpcode::kSort, shape, {dimension}), in HloSortInstruction()
1233 : HloDimensionsInstruction(HloOpcode::kTranspose, shape, dimensions) { in HloTransposeInstruction()
1256 : HloDimensionsInstruction(HloOpcode::kBroadcast, shape, in HloBroadcastInstruction()
|