Searched refs:kSqrt (Results 1 – 20 of 20) sorted by relevance
128 case OperatorType::kSqrt: in Run()288 unary_op->type == OperatorType::kSqrt || in Run()306 } else if (unary_op->type == OperatorType::kSqrt) { in Run()
46 if (sqrt_op->type != OperatorType::kSqrt) { in Run()
41 case OperatorType::kSqrt: in IsElementwiseOperator()
50 expected_op_type_producing_div_or_mul_input = OperatorType::kSqrt; in Run()
2099 case OperatorType::kSqrt: in Run()
137 V(kSqrt, "sqrt", 1) \
156 case HloOpcode::kSqrt: in IsExpensive()
710 case HloOpcode::kSqrt: in CreateUnary()1415 case HloOpcode::kSqrt: in CloneWithNewOperands()1770 case HloOpcode::kSqrt: in IdenticalSlowPath()2143 case HloOpcode::kSqrt: in IsElementwiseImpl()2591 case HloOpcode::kSqrt: in Visit()
95 opcode == HloOpcode::kPower || opcode == HloOpcode::kSqrt || in HandleElementwiseOp()
423 case HloOpcode::kSqrt: in EmitFloatUnaryOp()657 case HloOpcode::kSqrt: { in EmitComplexUnaryOp()2226 case HloOpcode::kSqrt: in MakeElementGenerator()
966 case HloOpcode::kSqrt: in GetInstructionColor()
2037 case HloOpcode::kSqrt: in InstructionCanChangeLayout()
259 case HloOpcode::kSqrt: in InferUnaryOpShape()
1046 HloInstruction::CreateUnary(divide->shape(), HloOpcode::kSqrt, b)); in HandleDivide()
751 case HloOpcode::kSqrt: in ParseInstructionRhs()
127 kSqrt, enumerator1354 TensorFlowSqrtOperator() : Operator(OperatorType::kSqrt) {}
2186 } else if (src_op.type == OperatorType::kSqrt) { in ConvertOperator()
139 CheckSimpleOperator<TensorFlowSqrtOperator>("SQRT", OperatorType::kSqrt); in TEST_F()
2546 "SQRT", OperatorType::kSqrt)); in BuildOperatorList()
3333 return operand.builder()->UnaryOp(HloOpcode::kSqrt, operand); in Sqrt()