| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dfs_hlo_visitor.cc | 31 HloOpcodeString(hlo->opcode())); in HandleElementwiseUnary()
38 HloOpcodeString(hlo->opcode())); in HandleElementwiseBinary()
|
| 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 | hlo_opcode.h | 173 string HloOpcodeString(HloOpcode opcode);
179 return os << HloOpcodeString(opcode);
|
| D | hlo_opcode.cc | 23 string HloOpcodeString(HloOpcode opcode) { in HloOpcodeString() function
|
| D | hlo_verifier.cc | 68 expected, HloOpcodeString(hlo->opcode()), in CheckOperandCount()
882 HloOpcodeString(sort->opcode()), sort->ToString()); in HandleSort()
1000 HloOpcodeString(reduce->opcode()), reduce->ToString()); in HandleReduce()
1788 << "The " << HloOpcodeString(instruction->opcode()) in VerifySingleUser()
1794 << "The consumer of a " << HloOpcodeString(instruction->opcode()) in VerifySingleUser()
1795 << " instruction needs to be " << HloOpcodeString(expected_user) in VerifySingleUser()
1796 << ", found " << HloOpcodeString(user->opcode()); in VerifySingleUser()
1803 << "The " << HloOpcodeString(instruction->opcode()) in VerifySingleOperand()
1809 << "The operand of a " << HloOpcodeString(instruction->opcode()) in VerifySingleOperand()
1810 << " instruction needs to be " << HloOpcodeString(expected_operand) in VerifySingleOperand()
[all …]
|
| D | shape_inference.cc | 262 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
281 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
294 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
306 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
323 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
340 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
367 HloOpcodeString(opcode)); in InferUnaryOpShape()
751 HloOpcodeString(operation), ShapeUtil::HumanString(lhs), in InferDegenerateDimensionBroadcastShape()
893 HloOpcodeString(operation), ShapeUtil::HumanString(lhs), in InferElementwiseBinaryOpShape()
951 HloOpcodeString(opcode), ShapeUtil::HumanStringWithLayout(lhs), in InferBinaryOpShape()
[all …]
|
| D | flatten_call_graph.cc | 65 << HloOpcodeString(instruction->opcode()); in ReplaceCalledComputation()
|
| D | local_service.cc | 85 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter) && in ParameterMetadata()
|
| D | hlo_instruction.cc | 524 << "but got " << HloOpcodeString(opcode); in CreateFromProto()
609 << "but got " << HloOpcodeString(opcode); in CreateFromProto()
1035 << HloOpcodeString(opcode); in CreateUnary()
1065 << HloOpcodeString(opcode); in CreateBinary()
1082 << HloOpcodeString(opcode); in CreateTernary()
2641 << HloOpcodeString(opcode()); in to_apply()
2664 << HloOpcodeString(opcode()); in set_to_apply()
2917 StrAppend(&result, HloOpcodeString(opcode()), "(", in ToStringWithCanonicalNameMap()
3160 return StrCat("%", name(), " = ", HloOpcodeString(opcode()), "(", in ToShortString()
3175 proto.set_opcode(HloOpcodeString(opcode_)); in ToProto()
[all …]
|
| D | elemental_ir_emitter.cc | 379 HloOpcodeString(op->opcode())); in EmitIntegerUnaryOp()
562 HloOpcodeString(op->opcode())); in EmitFloatUnaryOp()
852 HloOpcodeString(op->opcode())); in EmitComplexUnaryOp()
928 HloOpcodeString(op->opcode())); in EmitFloatBinaryOp()
1422 HloOpcodeString(op->opcode())); in EmitComplexBinaryOp()
1767 HloOpcodeString(op->opcode())); in EmitPredBinaryOp()
1771 HloOpcodeString(op->opcode())); in EmitPredBinaryOp()
1848 HloOpcodeString(op->opcode())); in EmitIntegerBinaryOp()
2715 HloOpcodeString(hlo->opcode())); in MakeElementGenerator()
|
| D | pattern_matcher.h | 1215 EXPLAIN << "HloInstruction has opcode " << HloOpcodeString(opcode_)
1221 << HloOpcodeString(opcode_);
1229 *os << "with opcode " << HloOpcodeString(opcode_);
1231 *os << "with any opcode other than " << HloOpcodeString(opcode_);
|
| D | conditional_code_motion.cc | 921 HloOpcodeString(op->opcode()) + "->" + in ReusesCarriedBy()
922 HloOpcodeString(user->opcode())) in ReusesCarriedBy()
974 "Move-" + HloOpcodeString(opcode)) in WorthHoisting()
|
| D | all_reduce_combiner_test.cc | 87 HloComputation::Builder sum_builder(HloOpcodeString(type)); in MakeReduction()
|
| D | hlo_graph_dumper.cc | 1112 if (absl::StartsWith(instr->name(), HloOpcodeString(instr->opcode()))) { in GetInstructionNodeLabel()
1116 StrCat(HloOpcodeString(instr->opcode()), in GetInstructionNodeLabel()
|
| D | hlo_parser.cc | 1051 HloOpcodeString(opcode))); in ParseInstructionRhs()
1088 HloOpcodeString(opcode), inferred.status().error_message())); in ParseInstructionRhs()
1415 << HloOpcodeString(opcode); in ParseInstructionRhs()
1437 << HloOpcodeString(opcode); in ParseInstructionRhs()
2716 HloOpcodeString(opcode))); in ParseInstructionRhs()
|
| D | buffer_assignment.cc | 213 HloOpcodeString(instruction->opcode())); in GatherComputationsByAllocationType()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | dumped_computation_to_operation_list.cc | 52 HloOpcodeString(hlo->opcode()), params, in DefaultAction()
|
| D | interactive_graphviz.cc | 327 std::cout << " Opcode: " << HloOpcodeString(instr->opcode()) << std::endl; in DoInfoCommand()
|
| D | replay_computation.cc | 150 if (instruction.opcode() == HloOpcodeString(is_infeed in GetXfeedShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_vectorization_test.cc | 50 std::string opcode = HloOpcodeString(spec.opcode); in Name()
|
| D | cpu_intrinsic_test.cc | 49 string opcode = HloOpcodeString(spec.opcode); in Name()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | utils.cc | 84 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter)) { in GetShardedProgramShapes()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 34 std::string op_name = HloOpcodeString(opcode); in GetMlirOpName()
423 HloOpcodeString(opcode)); in AddInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 280 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter)) { in GetProgramShape()
367 if (HloOpcodeString(HloOpcode::kTuple) == operand_instr.opcode()) { in IsConstantVisitor()
394 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter) && in SetDynamicBinding()
706 HloOpcodeString(binop)); in BinaryOp()
3555 HloOpcodeString(HloOpcode::kGetDimensionSize) || in BuildConstantSubGraph()
3561 HloOpcodeString(HloOpcode::kGetDimensionSize)) { in BuildConstantSubGraph()
3593 *const_instr.mutable_opcode() = HloOpcodeString(HloOpcode::kConstant); in BuildConstantSubGraph()
3601 HloOpcodeString(HloOpcode::kGetTupleElement)) { in BuildConstantSubGraph()
3607 if (maybe_tuple_instr->opcode() == HloOpcodeString(HloOpcode::kTuple)) { in BuildConstantSubGraph()
3643 if (instr_src->opcode() == HloOpcodeString(HloOpcode::kGetDimensionSize) || in BuildConstantSubGraph()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | instruction_fusion_test.cc | 417 computation.name(), HloOpcodeString(op)); in FindHloInstruction()
|