Home
last modified time | relevance | path

Searched refs:dot_info_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc192 DotInfo dot_info_; member in xla::cpu::__anond13f39450111::DotOpEmitter
214 : dot_info_(std::move(dot_info)), in DotOpEmitter()
226 PrimitiveType primitive_type = dot_info_.result_shape.element_type(); in EmitTiledLlvmIrGemm()
264 PrimitiveType primitive_type = dot_info_.result_shape.element_type(); in EmitTiledLlvmIrGemv()
387 switch (GetDotImplementationStrategy(hlo_module_config_, dot_info_, in Emit()
411 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm()
685 CHECK_EQ(dot_info_.result_shape.dimensions_size(), 2); in GetMatMultDims()
689 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims()