| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_unary.cc | 128 case OperatorType::kSqrt: in Run()
292 unary_op->type == OperatorType::kSqrt || in Run()
310 } else if (unary_op->type == OperatorType::kSqrt) { in Run()
|
| D | identify_l2_pool.cc | 32 if (sqrt_op->type != OperatorType::kSqrt) { in Run()
|
| D | reorder_elementwise_unary.cc | 42 case OperatorType::kSqrt: in IsElementwiseOperator()
|
| D | identify_l2_normalization.cc | 36 expected_op_type_producing_div_or_mul_input = OperatorType::kSqrt; in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | target_util.h | 62 kSqrt, enumerator
|
| D | target_util.cc | 153 case TargetDeviceFunctionID::kSqrt: { in GetDeviceFunctionRoot()
|
| D | elemental_ir_emitter.cc | 248 return EmitDeviceMathCall(TargetDeviceFunctionID::kSqrt, {value}, {prim_type}, in EmitSqrt()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | identify_l2_pool_test.cc | 70 EXPECT_FALSE(op->type == OperatorType::kSqrt); in RunIdentifyL2Pool()
|
| D | identify_l2_normalization_test.cc | 103 EXPECT_FALSE(op->type == OperatorType::kSqrt); in RunIdentifyL2Normalization()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode.h | 145 V(kSqrt, "sqrt", 1) \
|
| D | instruction_fusion.cc | 183 case HloOpcode::kSqrt: in IsExpensive()
|
| D | hlo_instruction.cc | 912 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()
|
| D | hlo_cost_analysis.cc | 106 opcode == HloOpcode::kSqrt || opcode == HloOpcode::kCbrt || in HandleElementwiseOp()
|
| D | elemental_ir_emitter.cc | 462 case HloOpcode::kSqrt: in EmitFloatUnaryOp()
792 case HloOpcode::kSqrt: { in EmitComplexUnaryOp()
2269 case HloOpcode::kSqrt: in MakeElementGenerator()
|
| D | hlo_graph_dumper.cc | 985 case HloOpcode::kSqrt: in GetInstructionColor()
|
| D | conditional_code_motion.cc | 193 case HloOpcode::kSqrt: in WorthHoisting()
|
| D | sharding_propagation.cc | 208 case HloOpcode::kSqrt: in PickRepresentativeOperand()
|
| D | algebraic_simplifier.cc | 117 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()
|
| D | hlo_parser.cc | 147 case HloOpcode::kSqrt: in CanInferShape()
1106 case HloOpcode::kSqrt: in ParseInstructionRhs()
|
| D | layout_assignment.cc | 2264 case HloOpcode::kSqrt: in InstructionCanChangeLayout()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 130 kSqrt, enumerator
1408 TensorFlowSqrtOperator() : Operator(OperatorType::kSqrt) {}
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 413 return xla::HloOpcode::kSqrt; in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 788 NoAttributeCase(kSqrt, SqrtOp); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 139 CheckSimpleOperator<TensorFlowSqrtOperator>("SQRT", OperatorType::kSqrt); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 376 case HloOpcode::kSqrt: in EmitOp()
|