Searched refs:DotInfo (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter_internal.h | 37 struct DotInfo { struct
43 explicit DotInfo(const HloInstruction& instr) { in DotInfo() argument
82 const HloModuleConfig& config, const DotInfo& dot_info,
|
| D | dot_op_emitter.cc | 68 struct DotInfo { struct
74 DotInfo() = default;
76 explicit DotInfo(const HloInstruction& instr) { in DotInfo() argument
118 const HloModuleConfig& config, const DotInfo& dot_info,
124 explicit DotOpEmitter(DotInfo dot_info, string dot_hlo_name,
221 DotInfo dot_info_;
236 DotInfo dot_info, string dot_hlo_name, const llvm_ir::IrArray& target_array, in DotOpEmitter()
961 bool IsAlignedGemm(const DotInfo& dot_info, in IsAlignedGemm()
973 const HloModuleConfig& config, const DotInfo& dot_info, in CanEmitTiledLlvmIrGemm()
1014 const HloModuleConfig& config, const DotInfo& dot_info, in GetDotImplementationStrategy()
[all …]
|