Searched refs:dot_info_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 215 dot_info_.result_shape.element_type()); in GetMlirGemmTileSize()
221 DotInfo dot_info_; member in xla::cpu::__anonb70c17dc0111::DotOpEmitter
242 : dot_info_(std::move(dot_info)), in DotOpEmitter()
255 Shape operand_shapes[] = {dot_info_.lhs_shape, dot_info_.rhs_shape}; in EmitLinalgMatmul()
261 int64 size_bytes = ShapeUtil::ByteSizeOf(dot_info_.result_shape); in EmitLinalgMatmul()
266 absl::StrCat("linalgMatMul_", dot_info_.result_shape.ToString(true), "_", in EmitLinalgMatmul()
267 dot_info_.lhs_shape.ToString(true), "_", in EmitLinalgMatmul()
268 dot_info_.rhs_shape.ToString(true)); in EmitLinalgMatmul()
271 mlir_context_, b_, dot_info_.result_shape, operand_shapes, target_ptr, in EmitLinalgMatmul()
273 CHECK_EQ(dot_info_.dim_nums.lhs_contracting_dimensions_size(), 1); in EmitLinalgMatmul()
[all …]
|