Searched refs:dot_info (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 118 const HloModuleConfig& config, const DotInfo& dot_info,
124 explicit DotOpEmitter(DotInfo dot_info, std::string dot_hlo_name,
247 DotInfo dot_info, std::string dot_hlo_name, in DotOpEmitter() argument
253 : dot_info_(std::move(dot_info)), in DotOpEmitter()
1144 bool IsAlignedGemm(const DotInfo& dot_info, in IsAlignedGemm() argument
1146 if (ShapeUtil::IsZeroElementArray(dot_info.lhs_shape) || in IsAlignedGemm()
1147 ShapeUtil::IsZeroElementArray(dot_info.rhs_shape)) { in IsAlignedGemm()
1151 return AreGemmShapes(dot_info.lhs_shape, dot_info.rhs_shape, in IsAlignedGemm()
1152 dot_info.result_shape, target_machine_features); in IsAlignedGemm()
1156 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,
|