Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dlogistic_expander_test.cc64 EXPECT_EQ(root->opcode(), HloOpcode::kLogistic); in TEST_F()
91 EXPECT_EQ(root->opcode(), HloOpcode::kLogistic); in TEST_F()
Dlogistic_expander.cc75 return instruction->opcode() == HloOpcode::kLogistic; in InstructionMatchesPattern()
Dhlo_opcode.h101 V(kLogistic, "logistic", 1) \
Dinstruction_fusion.cc166 case HloOpcode::kLogistic: in IsExpensive()
Dhlo_instruction.cc909 case HloOpcode::kLogistic: in CreateUnary()
1735 case HloOpcode::kLogistic: in CloneWithNewOperands()
2166 case HloOpcode::kLogistic: in IdenticalSlowPath()
2634 case HloOpcode::kLogistic: in IsOpElementwise()
3070 case HloOpcode::kLogistic: in Visit()
Dhlo_cost_analysis.cc105 opcode == HloOpcode::kLogistic || opcode == HloOpcode::kPower || in HandleElementwiseOp()
Dhlo_graph_dumper.cc980 case HloOpcode::kLogistic: in GetInstructionColor()
Dsharding_propagation.cc191 case HloOpcode::kLogistic: in PickRepresentativeOperand()
Dhlo_parser.cc119 case HloOpcode::kLogistic: in CanInferShape()
1098 case HloOpcode::kLogistic: in ParseInstructionRhs()
Dlayout_assignment.cc2236 case HloOpcode::kLogistic: in InstructionCanChangeLayout()
Dshape_inference.cc282 case HloOpcode::kLogistic: in InferUnaryOpShape()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_lstm.cc159 &state_output_tanh, OperatorType::kLogistic, in Run()
192 nullptr, OperatorType::kLogistic, in Run()
199 if (!MatchOperatorInputs(*state_remember_mul, *model, OperatorType::kLogistic, in Run()
Dquantize.cc68 OperatorType::kLogistic, in SupportsQuantization()
344 if (op.type == OperatorType::kLogistic || op.type == OperatorType::kSoftmax) { in ChooseHardcodedQuantizationForOperatorOutput()
Dhardcode_min_max.cc497 case OperatorType::kLogistic: in Run()
Dpropagate_fixed_sizes.cc2174 case OperatorType::kLogistic: in Run()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dop_version.cc196 {{OperatorType::kLogistic, 1}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
197 {{OperatorType::kLogistic, 2}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
198 {{OperatorType::kLogistic, 3}, kPendingReleaseOpVersion}, in GetMinimumRuntimeVersionForModel()
Doperator_test.cc120 CheckSimpleOperator<LogisticOperator>("LOGISTIC", OperatorType::kLogistic); in TEST_F()
Doperator.cc2011 ::tflite::BuiltinOperator_LOGISTIC, OperatorType::kLogistic)); in BuildOperatorList()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h67 kLogistic, enumerator
820 LogisticOperator() : Operator(OperatorType::kLogistic) {}
Dtooling_util.cc1961 case OperatorType::kLogistic: in EstimateArithmeticOpsCount()
Dexport_tensorflow.cc2177 } else if (src_op.type == OperatorType::kLogistic) { in ConvertOperator()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc394 return xla::HloOpcode::kLogistic; in MhloToHloOpcode()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3497 case HloOpcode::kLogistic: in BuildDynamicInferenceGraph()
4755 return operand.builder()->UnaryOp(HloOpcode::kLogistic, operand); in Logistic()