Searched refs:gemm_bias (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo_gpu_ops.mlir | 202 // CHECK-LABEL: func @gemm_bias
203 func @gemm_bias(%lhs: memref<5x4xf32>, %rhs: memref<4x5xf32>,
205 "lmhlo_gpu.gemm_bias"(%lhs, %rhs, %bias, %output) { dot_dimension_numbers = {
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_gpu_ops.td | 182 def LHLOGPU_GEMM_BiasOp : LHLOGPU_Op<"gemm_bias"> {
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 152 // CHECK: "lmhlo_gpu.gemm_bias"
177 // CHECK: "lmhlo_gpu.gemm_bias"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 693 auto gemm_bias, in EmitGemm()
695 gemm_bias.betaAttr(builder_.getF64FloatAttr(config.beta())); in EmitGemm()
696 return set_common_attributes(gemm_bias); in EmitGemm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1264 } else if (auto gemm_bias = in EmitGemmThunkFromMlir() local
1266 config = build_gemm_config(gemm_bias); in EmitGemmThunkFromMlir()
1267 config.backend_config.set_beta(gemm_bias.beta().convertToDouble()); in EmitGemmThunkFromMlir()
1268 TF_ASSIGN_OR_RETURN(lhs, GetAllocationSliceForMlir(gemm_bias.lhs())); in EmitGemmThunkFromMlir()
1269 TF_ASSIGN_OR_RETURN(rhs, GetAllocationSliceForMlir(gemm_bias.rhs())); in EmitGemmThunkFromMlir()
1270 TF_ASSIGN_OR_RETURN(bias, GetAllocationSliceForMlir(gemm_bias.bias())); in EmitGemmThunkFromMlir()
1271 TF_ASSIGN_OR_RETURN(output, GetAllocationSliceForMlir(gemm_bias.output())); in EmitGemmThunkFromMlir()
|