| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | depthwise_convolution_converter_test.cc | 60 << HloOpcodeString(root->opcode()) << " vs Reshape"; in TEST_F()
66 << HloOpcodeString(new_conv->opcode()) << " vs Convolution"; in TEST_F()
81 << HloOpcodeString(reshape_1->opcode()) << " vs Reshape"; in TEST_F()
86 << HloOpcodeString(transpose->opcode()) << " vs Transpose"; in TEST_F()
91 << HloOpcodeString(reshape_2->opcode()) << " vs Reshape"; in TEST_F()
|
| 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())); in HandleElementwiseUnary()
38 HloOpcodeString(hlo->opcode())); in HandleElementwiseBinary()
|
| D | hlo_opcode.h | 161 string HloOpcodeString(HloOpcode opcode);
167 return os << HloOpcodeString(opcode);
|
| D | hlo_opcode.cc | 23 string HloOpcodeString(HloOpcode opcode) { in HloOpcodeString() function
|
| D | shape_inference.cc | 249 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
266 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
279 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
291 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
308 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
325 HloOpcodeString(opcode), PrimitiveType_Name(shape.element_type())); in InferUnaryOpShape()
352 HloOpcodeString(opcode)); in InferUnaryOpShape()
742 HloOpcodeString(operation), ShapeUtil::HumanString(lhs), in InferDegenerateDimensionBroadcastShape()
884 HloOpcodeString(operation), ShapeUtil::HumanString(lhs), in InferElementwiseBinaryOpShape()
942 HloOpcodeString(opcode), ShapeUtil::HumanStringWithLayout(lhs), in InferBinaryOpShape()
[all …]
|
| D | flatten_call_graph.cc | 65 << HloOpcodeString(instruction->opcode()); in ReplaceCalledComputation()
|
| D | hlo_verifier.cc | 60 expected, HloOpcodeString(hlo->opcode()), in CheckOperandCount()
440 HloOpcodeString(sort->opcode()), sort->ToString()); in HandleSort()
558 HloOpcodeString(reduce->opcode()), reduce->ToString()); in HandleReduce()
1319 << HloOpcodeString(copy_done->opcode()); in VerifyAsynchronousCopies()
1330 << HloOpcodeString(copy_start->opcode()); in VerifyAsynchronousCopies()
1584 HloOpcodeString(instruction->opcode()), in CheckElementwiseInstruction()
|
| D | local_service.cc | 84 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter) && in ParameterMetadata()
|
| D | hlo_instruction.cc | 793 << HloOpcodeString(opcode); in CreateUnary()
823 << HloOpcodeString(opcode); in CreateBinary()
840 << HloOpcodeString(opcode); in CreateTernary()
2142 << HloOpcodeString(opcode()); in to_apply()
2163 << HloOpcodeString(opcode()); in set_to_apply()
2410 StrAppend(&result, " ", HloOpcodeString(opcode()), "(", in ToStringWithCanonicalNameMap()
2609 return StrCat("%", name(), " = ", HloOpcodeString(opcode()), "(", in ToShortString()
2624 proto.set_opcode(HloOpcodeString(opcode_)); in ToProto()
2665 return HloOpcodeString(opcode()); in ToCategory()
2722 name_(HloOpcodeString(opcode)) { in HloInstruction()
[all …]
|
| D | elemental_ir_emitter.cc | 362 HloOpcodeString(op->opcode())); in EmitIntegerUnaryOp()
511 HloOpcodeString(op->opcode())); in EmitFloatUnaryOp()
799 HloOpcodeString(op->opcode())); in EmitComplexUnaryOp()
872 HloOpcodeString(op->opcode())); in EmitFloatBinaryOp()
1292 HloOpcodeString(op->opcode())); in EmitComplexBinaryOp()
1590 HloOpcodeString(op->opcode())); in EmitIntegerBinaryOp()
2384 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 | hlo_graph_dumper.cc | 1097 if (absl::StartsWith(instr->name(), HloOpcodeString(instr->opcode()))) { in GetInstructionNodeLabel()
1101 StrCat(HloOpcodeString(instr->opcode()), in GetInstructionNodeLabel()
|
| D | hlo_evaluator_typed_visitor.h | 119 "Unsupported type for %s: %s", HloOpcodeString(instruction->opcode()),
176 HloOpcodeString(hlo_instruction->opcode()));
|
| D | buffer_assignment.cc | 204 HloOpcodeString(instruction->opcode())); in GatherComputationsByAllocationType()
|
| D | hlo_instruction_test.cc | 57 HloOpcodeString(hlo_instruction->opcode())); in DefaultAction()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | dumped_computation_to_operation_list.cc | 52 HloOpcodeString(hlo->opcode()), params, in DefaultAction()
|
| D | interactive_graphviz.cc | 307 std::cout << " Opcode: " << HloOpcodeString(instr->opcode()) << std::endl; in DoInfoCommand()
|
| D | replay_computation.cc | 141 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/service/mlir_gpu/ |
| D | hlo_dialect_emitter.cc | 87 "HLO Opcode ", HloOpcodeString(opcode), " is not supported.")); in InsertMlirOp()
|
| D | lhlo_dialect_emitter.cc | 118 "LHLO opcode ", HloOpcodeString(opcode), " is not supported.")); in InsertMlirOp()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 221 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter)) { in GetProgramShape()
305 if (instr.opcode() == HloOpcodeString(HloOpcode::kParameter) && in SetDynamicBinding()
567 HloOpcodeString(binop)); in BinaryOp()
2606 HloOpcodeString(HloOpcode::kGetDimensionSize)) { in BuildConstantSubGraph()
2630 *const_instr.mutable_opcode() = HloOpcodeString(HloOpcode::kConstant); in BuildConstantSubGraph()
2654 if (instr_src->opcode() == HloOpcodeString(HloOpcode::kGetDimensionSize)) { in BuildConstantSubGraph()
2768 instr.set_opcode(HloOpcodeString(opcode)); in AddInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | instruction_fusion_test.cc | 415 computation.name(), HloOpcodeString(op)); in FindHloInstruction()
|