Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_exporter.cc62 llvm::Optional<mlir::lmhlo_gpu::Activation> activation = in ConvertConvActivationMode()
63 mlir::lmhlo_gpu::symbolizeActivation(input); in ConvertConvActivationMode()
69 case mlir::lmhlo_gpu::Activation::None: in ConvertConvActivationMode()
71 case mlir::lmhlo_gpu::Activation::Sigmoid: in ConvertConvActivationMode()
73 case mlir::lmhlo_gpu::Activation::Tanh: in ConvertConvActivationMode()
75 case mlir::lmhlo_gpu::Activation::Relu: in ConvertConvActivationMode()
77 case mlir::lmhlo_gpu::Activation::Relu6: in ConvertConvActivationMode()
79 case mlir::lmhlo_gpu::Activation::ReluX: in ConvertConvActivationMode()
81 case mlir::lmhlo_gpu::Activation::BandPass: in ConvertConvActivationMode()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir8 …"lmhlo_gpu.batch_norm_grad"(%arg0, %arg1, %arg2, %arg3, %arg4, %grad_operand, %grad_scale, %grad_o…
17 …"lmhlo_gpu.batch_norm_inference"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg_out) {epsilon = 1.000000e…
26 …"lmhlo_gpu.batch_norm_training"(%arg0, %arg1, %arg2, %output, %batch_mean, %batch_var) {epsilon = …
36 "lmhlo_gpu.conv_forward"(%input, %filter, %output, %scratch)
65 "lmhlo_gpu.conv_backwardfilter"(%input, %filter, %output, %scratch)
95 "lmhlo_gpu.conv_backwardinput"(%input, %filter, %output, %scratch)
126 "lmhlo_gpu.conv_forward_fused"(%input, %filter, %bias, %output, %scratch)
157 …"lmhlo_gpu.conv_forward_fused_with_side_input"(%input, %filter, %bias, %side_input, %output, %scra…
188 "lmhlo_gpu.gemm"(%lhs, %rhs, %output) { dot_dimension_numbers = {
205 "lmhlo_gpu.gemm_bias"(%lhs, %rhs, %bias, %output) { dot_dimension_numbers = {
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc160 mlir::lmhlo::LmhloDialect, mlir::lmhlo_gpu::LmhloGpuDialect>(); in getDependentDialects()
650 StatusOr<lmhlo_gpu::CholeskyOp> LhloDialectEmitter::EmitCholesky( in EmitCholesky()
653 CreateOpWithoutAttrs<lmhlo_gpu::CholeskyOp>(custom_call)); in EmitCholesky()
687 CreateOpWithoutAttrs<lmhlo_gpu::GEMMOp>(custom_call)); in EmitGemm()
694 CreateOpWithoutAttrs<lmhlo_gpu::GEMM_BiasOp>(custom_call)); in EmitGemm()
702 static StatusOr<mlir::lmhlo_gpu::Activation> GetLHLOActivation( in GetLHLOActivation()
706 return mlir::lmhlo_gpu::Activation::None; in GetLHLOActivation()
708 return mlir::lmhlo_gpu::Activation::Sigmoid; in GetLHLOActivation()
710 return mlir::lmhlo_gpu::Activation::Relu; in GetLHLOActivation()
712 return mlir::lmhlo_gpu::Activation::Relu6; in GetLHLOActivation()
[all …]
Dmhlo_to_lhlo_with_xla.h66 xla::StatusOr<lmhlo_gpu::CholeskyOp> EmitCholesky(
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_gpu_ops_base.td24 let name = "lmhlo_gpu";
25 let cppNamespace = "::mlir::lmhlo_gpu";
Dlhlo_gpu_ops.h46 namespace lmhlo_gpu {
Dlhlo_gpu_ops_enums.td34 let cppNamespace = "::mlir::lmhlo_gpu";
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dlhlo_gpu_ops.cc48 namespace lmhlo_gpu { namespace
Dinit.cc27 mlir::lmhlo_gpu::LmhloGpuDialect>(); in registerAllMhloDialects()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt114 // CHECK: "lmhlo_gpu.cholesky"
129 // CHECK: "lmhlo_gpu.gemm"
152 // CHECK: "lmhlo_gpu.gemm_bias"
177 // CHECK: "lmhlo_gpu.gemm_bias"
277 // CHECK: "lmhlo_gpu.conv_forward"
313 // CHECK: "lmhlo_gpu.conv_forward_fused"
352 // CHECK: "lmhlo_gpu.conv_forward_fused_with_side_input"
394 // CHECK: "lmhlo_gpu.batch_norm_training"
415 // CHECK: "lmhlo_gpu.batch_norm_grad"
437 // CHECK: "lmhlo_gpu.batch_norm_inference"
/external/tensorflow/tensorflow/compiler/mlir/hlo/tools/mlir-hlo-opt/
Dmlir-hlo-opt.cpp35 registry.insert<mlir::lmhlo_gpu::LmhloGpuDialect>(); in main()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1081 if (isa<mlir::lmhlo_gpu::GEMMOp, mlir::lmhlo_gpu::GEMM_BiasOp>(input.op)) { in EmitCustomCallFromMlir()
1085 if (mlir::isa<mlir::lmhlo_gpu::ConvForwardOp, in EmitCustomCallFromMlir()
1086 mlir::lmhlo_gpu::ConvForwardFusedOp, in EmitCustomCallFromMlir()
1087 mlir::lmhlo_gpu::ConvForwardFusedSideInputOp, in EmitCustomCallFromMlir()
1088 mlir::lmhlo_gpu::ConvBackwardFilterOp, in EmitCustomCallFromMlir()
1089 mlir::lmhlo_gpu::ConvBackwardInputOp>(input.op)) { in EmitCustomCallFromMlir()
1093 if (isa<mlir::lmhlo_gpu::BatchNormTrainingOp, in EmitCustomCallFromMlir()
1094 mlir::lmhlo_gpu::BatchNormInferenceOp, in EmitCustomCallFromMlir()
1095 mlir::lmhlo_gpu::BatchNormGradOp>(input.op)) { in EmitCustomCallFromMlir()
1100 if (mlir::isa<mlir::lmhlo_gpu::CholeskyOp>(input.op)) { in EmitCustomCallFromMlir()
[all …]
Dgpu_compiler.cc594 mlir::lmhlo_gpu::LmhloGpuDialect>(); in CompileModuleToLlvmIrImpl()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc148 mlir::lmhlo_gpu::LmhloGpuDialect>(); in ParseMlirModule()