Home
last modified time | relevance | path

Searched refs:HloOpcodeString (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_verifier.cc77 expected, HloOpcodeString(hlo->opcode()), in CheckOperandCount()
899 HloOpcodeString(sort->opcode()), sort->ToString()); in HandleSort()
1026 HloOpcodeString(reduce->opcode()), reduce->ToString()); in HandleReduce()
1361 HloOpcodeString(async_op->opcode()), in CheckAsyncOpOperand()
1362 HloOpcodeString(operand->opcode())); in CheckAsyncOpOperand()
1370 HloOpcodeString(async_op->opcode()), in CheckAsyncOpOperand()
1379 HloOpcodeString(async_op->opcode()), in CheckAsyncOpOperand()
1394 HloOpcodeString(async_op->opcode()), async_shape.ToString()); in CheckAsyncOpComputationShapes()
1403 HloOpcodeString(async_op->opcode()), in CheckAsyncOpComputationShapes()
1411 HloOpcodeString(async_op->opcode()), in CheckAsyncOpComputationShapes()
[all …]
H A Ddfs_hlo_visitor.cc31 HloOpcodeString(hlo->opcode())); in HandleElementwiseUnary()
38 HloOpcodeString(hlo->opcode())); in HandleElementwiseBinary()
H A Dhlo_opcode_test.cc26 ASSERT_EQ("multiply", HloOpcodeString(HloOpcode::kMultiply)); in TEST()
41 EXPECT_EQ(opcode, StringToHloOpcode(HloOpcodeString(opcode)).ValueOrDie()); in TEST()
H A Dshape_inference.cc254 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
273 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
286 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
298 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
315 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
332 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
359 HloOpcodeString(opcode)); in InferUnaryOpShape()
766 HloOpcodeString(operation), ShapeUtil::HumanString(lhs), in InferDegenerateDimensionBroadcastShape()
908 HloOpcodeString(operation), ShapeUtil::HumanString(lhs), in InferElementwiseBinaryOpShape()
966 HloOpcodeString(opcode), ShapeUtil::HumanStringWithLayout(lhs), in InferBinaryOpShape()
[all …]
H A Dhlo_opcode.h181 std::string HloOpcodeString(HloOpcode opcode);
187 return os << HloOpcodeString(opcode);
H A Dhlo_instruction.cc580 << "but got " << HloOpcodeString(opcode); in CreateFromProto()
665 << "but got " << HloOpcodeString(opcode); in CreateFromProto()
1106 << HloOpcodeString(opcode); in CreateUnary()
1136 << HloOpcodeString(opcode); in CreateBinary()
1152 << HloOpcodeString(opcode); in CreateTernary()
2802 << HloOpcodeString(opcode()); in to_apply()
2825 << HloOpcodeString(opcode()); in set_to_apply()
3092 << "Unexpected async opcode: " << HloOpcodeString(opcode()); in ToStringWithCanonicalNameMap()
3096 StrAppend(&result, HloOpcodeString(async_wrapped_opcode()), suffix); in ToStringWithCanonicalNameMap()
3098 StrAppend(&result, HloOpcodeString(opcode())); in ToStringWithCanonicalNameMap()
[all …]
H A Dhlo_opcode.cc24 std::string HloOpcodeString(HloOpcode opcode) { in HloOpcodeString() function
H A Dflatten_call_graph.cc74 << HloOpcodeString(instruction->opcode()); in ReplaceCalledComputation()
H A Delemental_ir_emitter.cc370 HloOpcodeString(op->opcode())); in EmitIntegerUnaryOp()
560 HloOpcodeString(op->opcode())); in EmitFloatUnaryOp()
850 HloOpcodeString(op->opcode())); in EmitComplexUnaryOp()
931 HloOpcodeString(op->opcode())); in EmitFloatBinaryOp()
1425 HloOpcodeString(op->opcode())); in EmitComplexBinaryOp()
1770 HloOpcodeString(op->opcode())); in EmitPredBinaryOp()
1774 HloOpcodeString(op->opcode())); in EmitPredBinaryOp()
1851 HloOpcodeString(op->opcode())); in EmitIntegerBinaryOp()
2724 HloOpcodeString(hlo->opcode())); in MakeElementGenerator()
H A Dlocal_service.cc84 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter) && in ParameterMetadata()
H A Dpattern_matcher.h1244 EXPLAIN << "HloInstruction has opcode " << HloOpcodeString(opcode_)
1250 << HloOpcodeString(opcode_);
1258 *os << "with opcode " << HloOpcodeString(opcode_);
1260 *os << "with any opcode other than " << HloOpcodeString(opcode_);
H A Dconditional_code_motion.cc1015 HloOpcodeString(op->opcode()) + "->" + in ReusesCarriedBy()
1016 HloOpcodeString(user->opcode())) in ReusesCarriedBy()
1069 "Move-" + HloOpcodeString(opcode)) in WorthHoisting()
H A Dall_reduce_combiner_test.cc85 HloComputation::Builder sum_builder(HloOpcodeString(type)); in MakeReduction()
H A Dhlo_instructions.cc1555 SetAndSanitizeName(HloOpcodeString(opcode)); in HloCallableInstruction()
1566 SetAndSanitizeName(std::string(prefix) + HloOpcodeString(opcode)); in HloCallableInstruction()
1578 SetAndSanitizeName(HloOpcodeString(opcode)); in HloCallableInstruction()
1817 SetAndSanitizeName(HloOpcodeString(opcode())); in HloFusionInstruction()
2147 SetAndSanitizeName(HloOpcodeString(opcode())); in HloCallInstruction()
H A Dhlo_graph_dumper.cc1181 if (absl::StartsWith(instr->name(), HloOpcodeString(instr->opcode()))) { in GetInstructionNodeLabel()
1185 StrCat(HloOpcodeString(instr->opcode()), in GetInstructionNodeLabel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/
H A Ddumped_computation_to_operation_list.cc54 HloOpcodeString(hlo->opcode()), params, in DefaultAction()
H A Dinteractive_graphviz.cc337 std::cout << " Opcode: " << HloOpcodeString(instr->opcode()) << std::endl; in DoInfoCommand()
H A Dreplay_computation.cc150 if (instruction.opcode() == HloOpcodeString(is_infeed in GetXfeedShape()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dhlo_fusion_stats.cc73 opcodes_.insert(HloOpcodeString(instr->opcode())); in DefaultAction()
H A Dinstruction_fusion_test.cc417 computation.name(), HloOpcodeString(op)); in FindHloInstruction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
H A Dcpu_vectorization_test.cc50 std::string opcode = HloOpcodeString(spec.opcode); in Name()
H A Dcpu_intrinsic_test.cc50 std::string opcode = HloOpcodeString(spec.opcode); in Name()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc333 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter)) { in GetProgramShape()
420 if (HloOpcodeString(HloOpcode::kTuple) == operand_instr.opcode()) { in IsConstantVisitor()
447 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter) && in SetDynamicBinding()
762 HloOpcodeString(binop)); in BinaryOp()
3698 HloOpcodeString(HloOpcode::kGetDimensionSize) || in BuildConstantSubGraph()
3704 HloOpcodeString(HloOpcode::kGetDimensionSize)) { in BuildConstantSubGraph()
3736 *const_instr.mutable_opcode() = HloOpcodeString(HloOpcode::kConstant); in BuildConstantSubGraph()
3744 HloOpcodeString(HloOpcode::kGetTupleElement)) { in BuildConstantSubGraph()
3750 if (maybe_tuple_instr->opcode() == HloOpcodeString(HloOpcode::kTuple)) { in BuildConstantSubGraph()
3786 if (instr_src->opcode() == HloOpcodeString(HloOpcode::kGetDimensionSize) || in BuildConstantSubGraph()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dutils.cc86 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter)) { in GetShardedProgramShapes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
H A Dmlir_hlo_builder.cc41 std::string op_name = HloOpcodeString(opcode); in GetMlirOpName()
527 HloOpcodeString(opcode)); in AddInstruction()

12