Searched refs:HloOpcodeString (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode_test.cc | 26 ASSERT_EQ("multiply", HloOpcodeString(HloOpcode::kMultiply)); in TEST()
41 EXPECT_EQ(opcode, StringToHloOpcode(HloOpcodeString(opcode)).ValueOrDie()); in TEST()
|
| D | dfs_hlo_visitor.cc | 31 HloOpcodeString(hlo->opcode()).c_str()); in HandleElementwiseUnary()
38 HloOpcodeString(hlo->opcode()).c_str()); in HandleElementwiseBinary()
|
| D | hlo_opcode.h | 147 string HloOpcodeString(HloOpcode opcode);
153 return os << HloOpcodeString(opcode);
|
| D | hlo_opcode.cc | 23 string HloOpcodeString(HloOpcode opcode) { in HloOpcodeString() function
|
| D | flatten_call_graph.cc | 62 << HloOpcodeString(instruction->opcode()); in ReplaceCalledComputation()
|
| D | hlo_instruction.cc | 244 << HloOpcodeString(opcode); in CreateUnary()
280 << HloOpcodeString(opcode); in CreateBinary()
296 << HloOpcodeString(opcode); in CreateTernary()
1438 LOG(FATAL) << "Not yet implemented, clone: " << HloOpcodeString(opcode_); in CloneWithNewOperands()
1975 << HloOpcodeString(opcode()); in to_apply()
1993 << HloOpcodeString(opcode()); in set_to_apply()
2105 HloOpcodeString(opcode()), "(", OperandsToString(options), ")"); in ToString()
2303 return StrCat("%", name(), " = ", HloOpcodeString(opcode()), "(", in ToShortString()
2314 proto.set_opcode(HloOpcodeString(opcode_)); in ToProto()
2433 return HloOpcodeString(opcode()); in ToCategory()
[all …]
|
| D | elemental_ir_emitter.cc | 332 HloOpcodeString(op->opcode()).c_str()); in EmitIntegerUnaryOp()
458 HloOpcodeString(op->opcode()).c_str()); in EmitFloatUnaryOp()
655 HloOpcodeString(op->opcode()).c_str()); in EmitComplexUnaryOp()
727 HloOpcodeString(op->opcode()).c_str()); in EmitFloatBinaryOp()
856 HloOpcodeString(op->opcode()).c_str()); in EmitComplexBinaryOp()
1061 HloOpcodeString(op->opcode()).c_str()); in EmitIntegerBinaryOp()
1858 HloOpcodeString(hlo->opcode()).c_str()); in MakeElementGenerator()
|
| D | hlo_tfgraph_builder.cc | 39 string name = StrCat("hlo-", HloOpcodeString(instruction->opcode())); in GetOpDefName()
|
| D | hlo_graph_dumper.cc | 1019 .starts_with(HloOpcodeString(instr->opcode()))) { in GetInstructionNodeLabel()
1023 StrCat(HloOpcodeString(instr->opcode()), in GetInstructionNodeLabel()
|
| D | hlo_module.cc | 247 if (instruction.opcode() == HloOpcodeString(HloOpcode::kParameter)) { in ProgramShapeFromProto()
|
| D | hlo_evaluator.cc | 98 << HloOpcodeString(opcode); in Compare()
129 << HloOpcodeString(opcode); in Compare()
232 HloOpcodeString(hlo_instruction->opcode()).c_str()); in DefaultAction()
|
| D | buffer_assignment.cc | 614 HloOpcodeString(instruction->opcode()).c_str()); in GatherComputationsByAllocationType()
|
| D | hlo_instruction_test.cc | 53 HloOpcodeString(hlo_instruction->opcode()).c_str()); in DefaultAction()
|
| D | hlo_dataflow_analysis_test.cc | 83 HloComputation::Builder builder(TestName() + "." + HloOpcodeString(opcode)); in CreateR0F32UnaryOpComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | while_transformer.cc | 126 HloOpcodeString(instruction->opcode()).c_str(), in Match()
127 HloOpcodeString(opcode_).c_str()); in Match()
130 VLOG(2) << "Matched " << HloOpcodeString(opcode_) << ": " << tag_; in Match()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | dumped_computation_to_operation_list.cc | 54 "%s :: (%s) -> %s :: %s\n", HloOpcodeString(hlo->opcode()).c_str(), in DefaultAction()
|
| /external/tensorflow/tensorflow/compiler/xla/legacy_flags/ |
| D | debug_options_parsers.h | 121 if (opcode == HloOpcodeString(static_cast<HloOpcode>(i))) { in parse_xla_reduce_precision_option()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| D | hlo_parser.cc | 1057 HloOpcodeString(opcode))); in ParseInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2927 HloOpcodeString(instruction.opcode()).c_str()); in ElementTypesSameAndSupported()
|