Home
last modified time | relevance | path

Searched refs:kSqrt (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_unary.cc128 case OperatorType::kSqrt: in Run()
292 unary_op->type == OperatorType::kSqrt || in Run()
310 } else if (unary_op->type == OperatorType::kSqrt) { in Run()
Didentify_l2_pool.cc32 if (sqrt_op->type != OperatorType::kSqrt) { in Run()
Dreorder_elementwise_unary.cc42 case OperatorType::kSqrt: in IsElementwiseOperator()
Didentify_l2_normalization.cc36 expected_op_type_producing_div_or_mul_input = OperatorType::kSqrt; in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_util.h62 kSqrt, enumerator
Dtarget_util.cc153 case TargetDeviceFunctionID::kSqrt: { in GetDeviceFunctionRoot()
Delemental_ir_emitter.cc248 return EmitDeviceMathCall(TargetDeviceFunctionID::kSqrt, {value}, {prim_type}, in EmitSqrt()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Didentify_l2_pool_test.cc70 EXPECT_FALSE(op->type == OperatorType::kSqrt); in RunIdentifyL2Pool()
Didentify_l2_normalization_test.cc103 EXPECT_FALSE(op->type == OperatorType::kSqrt); in RunIdentifyL2Normalization()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_opcode.h145 V(kSqrt, "sqrt", 1) \
Dinstruction_fusion.cc183 case HloOpcode::kSqrt: in IsExpensive()
Dhlo_instruction.cc912 case HloOpcode::kSqrt: in CreateUnary()
1738 case HloOpcode::kSqrt: in CloneWithNewOperands()
2169 case HloOpcode::kSqrt: in IdenticalSlowPath()
2637 case HloOpcode::kSqrt: in IsOpElementwise()
3180 case HloOpcode::kSqrt: in Visit()
Dhlo_cost_analysis.cc106 opcode == HloOpcode::kSqrt || opcode == HloOpcode::kCbrt || in HandleElementwiseOp()
Delemental_ir_emitter.cc462 case HloOpcode::kSqrt: in EmitFloatUnaryOp()
792 case HloOpcode::kSqrt: { in EmitComplexUnaryOp()
2269 case HloOpcode::kSqrt: in MakeElementGenerator()
Dhlo_graph_dumper.cc985 case HloOpcode::kSqrt: in GetInstructionColor()
Dconditional_code_motion.cc193 case HloOpcode::kSqrt: in WorthHoisting()
Dsharding_propagation.cc208 case HloOpcode::kSqrt: in PickRepresentativeOperand()
Dalgebraic_simplifier.cc117 case HloOpcode::kSqrt: in IsPositive()
1634 HloInstruction::CreateUnary(divide->shape(), HloOpcode::kSqrt, b)); in HandleDivide()
4334 rsqrt, HloInstruction::CreateUnary(rsqrt->shape(), HloOpcode::kSqrt, in HandleRsqrt()
Dhlo_parser.cc147 case HloOpcode::kSqrt: in CanInferShape()
1106 case HloOpcode::kSqrt: in ParseInstructionRhs()
Dlayout_assignment.cc2264 case HloOpcode::kSqrt: in InstructionCanChangeLayout()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h130 kSqrt, enumerator
1408 TensorFlowSqrtOperator() : Operator(OperatorType::kSqrt) {}
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc413 return xla::HloOpcode::kSqrt; in MhloToHloOpcode()
Dhlo_function_importer.cc788 NoAttributeCase(kSqrt, SqrtOp); in ImportInstructionImpl()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc139 CheckSimpleOperator<TensorFlowSqrtOperator>("SQRT", OperatorType::kSqrt); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc376 case HloOpcode::kSqrt: in EmitOp()

12