Searched refs:dot_info (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 103 const HloModuleConfig& config, const DotInfo& dot_info,
109 explicit DotOpEmitter(DotInfo dot_info, string dot_hlo_name,
205 DotOpEmitter::DotOpEmitter(DotInfo dot_info, string dot_hlo_name, in DotOpEmitter() argument
214 : dot_info_(std::move(dot_info)), in DotOpEmitter()
771 bool IsAlignedGemm(const DotInfo& dot_info, in IsAlignedGemm() argument
773 if (ShapeUtil::IsZeroElementArray(dot_info.lhs_shape) || in IsAlignedGemm()
774 ShapeUtil::IsZeroElementArray(dot_info.rhs_shape)) { in IsAlignedGemm()
778 return AreGemmShapes(dot_info.lhs_shape, dot_info.rhs_shape, in IsAlignedGemm()
779 dot_info.result_shape, target_machine_features); in IsAlignedGemm()
783 const HloModuleConfig& config, const DotInfo& dot_info, in CanEmitTiledLlvmIrGemm() argument
[all …]
|
| D | dot_op_emitter_internal.h | 82 const HloModuleConfig& config, const DotInfo& dot_info,
|