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 | 56 struct DotInfo { struct
62 DotInfo() = default;
64 explicit DotInfo(const HloInstruction& instr) { in DotInfo() function
103 const HloModuleConfig& config, const DotInfo& dot_info,
109 explicit DotOpEmitter(DotInfo dot_info, string dot_hlo_name,
192 DotInfo dot_info_;
205 DotOpEmitter::DotOpEmitter(DotInfo dot_info, string dot_hlo_name, in DotOpEmitter()
771 bool IsAlignedGemm(const DotInfo& dot_info, in IsAlignedGemm()
783 const HloModuleConfig& config, const DotInfo& dot_info, in CanEmitTiledLlvmIrGemm()
822 const HloModuleConfig& config, const DotInfo& dot_info, in GetDotImplementationStrategy()
[all …]
|