Searched refs:LogisticOp (Results 1 – 14 of 14) sorted by relevance
| /external/gemmlowp/test/ |
| D | test_fixedpoint.cc | 305 class LogisticOp final : public UnaryOpBase { class in gemmlowp::__anon843601290111::TestFixedPoint
536 TestUnaryOp(LogisticOp<0>(), testvals); in RunTests()
537 TestUnaryOp(LogisticOp<1>(), testvals); in RunTests()
538 TestUnaryOp(LogisticOp<2>(), testvals); in RunTests()
539 TestUnaryOp(LogisticOp<3>(), testvals); in RunTests()
540 TestUnaryOp(LogisticOp<4>(), testvals); in RunTests()
541 TestUnaryOp(LogisticOp<5>(), testvals); in RunTests()
542 TestUnaryOp(LogisticOp<6>(), testvals); in RunTests()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo/transforms/ |
| D | map_hlo_to_lhlo_op.h | 74 MAP_HLO_TO_LHLO(LogisticOp);
|
| D | map_lhlo_to_hlo_op.h | 72 MAP_LHLO_TO_HLO(LogisticOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/hardwares/ |
| D | gpu_hardware.cc | 173 TAC_REGISTER_GPU_OP(LogisticOp, CreateBasicOpNoCost);
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/ |
| D | gpu_fusion_rewrite.cc | 344 mhlo::Log1pOp, mhlo::LogisticOp, mhlo::MulOp, mhlo::NegOp, mhlo::RoundOp, in getRewritableTarget()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | hlo_legalize_to_lhlo.cc | 541 HloToLhloOpConverter<mhlo::LogisticOp>, in populateHloToLhloConversionPattern()
|
| D | legalize_to_linalg.cc | 3300 PointwiseToLinalgConverter<mhlo::LogisticOp>, in populateHloToLinalgConversionPattern()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 390 } else if (isa<mlir::mhlo::LogisticOp>(op)) { in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 1458 NO_ATTRIBUTE_CASE(kLogistic, LogisticOp); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | map_mhlo_to_scalar_op.h | 822 inline Value mapMhloOpToStdScalarOp<mhlo::LogisticOp>(
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 3682 int64_t LogisticOp::GetArithmeticCount(Operation *op) { in GetArithmeticCount()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 2632 auto tfl_logistic_op = cast<TFL::LogisticOp>(op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| D | StablehloOps.cpp | 694 INFER_RETURN_TYPE_COMPONENTS_FROM_OPERANDS(LogisticOp) in INFER_RETURN_TYPE_COMPONENTS_FROM_OPERANDS()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 692 INFER_RETURN_TYPE_COMPONENTS_FROM_OPERANDS(LogisticOp) in INFER_RETURN_TYPE_COMPONENTS_FROM_OPERANDS()
|