HomeSort by relevance Sort by last modified time
    Searched refs:dot_info (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc 101 // `dot_info`.
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,
214 : dot_info_(std::move(dot_info)),
771 bool IsAlignedGemm(const DotInfo& dot_info,
773 if (ShapeUtil::IsZeroElementArray(dot_info.lhs_shape) ||
774 ShapeUtil::IsZeroElementArray(dot_info.rhs_shape)) {
778 return AreGemmShapes(dot_info.lhs_shape, dot_info.rhs_shape
965 DotInfo dot_info; local
    [all...]
dot_op_emitter_internal.h 80 // `dot_info`.
82 const HloModuleConfig& config, const DotInfo& dot_info,

Completed in 112 milliseconds