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, string dot_hlo_name,
236 DotInfo dot_info, string dot_hlo_name, const llvm_ir::IrArray& target_array, in DotOpEmitter() argument
242 : dot_info_(std::move(dot_info)), in DotOpEmitter()
961 bool IsAlignedGemm(const DotInfo& dot_info, in IsAlignedGemm() argument
963 if (ShapeUtil::IsZeroElementArray(dot_info.lhs_shape) || in IsAlignedGemm()
964 ShapeUtil::IsZeroElementArray(dot_info.rhs_shape)) { in IsAlignedGemm()
968 return AreGemmShapes(dot_info.lhs_shape, dot_info.rhs_shape, in IsAlignedGemm()
969 dot_info.result_shape, target_machine_features); in IsAlignedGemm()
973 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,
|