| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_unary.cc | 127 case OperatorType::kRsqrt: in Run()
291 unary_op->type == OperatorType::kRsqrt || in Run()
308 } else if (unary_op->type == OperatorType::kRsqrt) { in Run()
|
| D | identify_l2_normalization.cc | 38 expected_op_type_producing_div_or_mul_input = OperatorType::kRsqrt; in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | target_util.h | 60 kRsqrt, enumerator
|
| D | instruction_fusion.cc | 46 case HloOpcode::kRsqrt: in IsExpensive()
|
| D | target_util.cc | 147 case TargetDeviceFunctionID::kRsqrt: { in GetDeviceFunctionRoot()
|
| D | horizontal_loop_fusion_test.cc | 341 HloInstruction::CreateUnary(shape, HloOpcode::kRsqrt, ms_eps)); in TEST_F()
|
| D | cudnn_batchnorm_rewriter.cc | 243 var_plus_epsilon->shape(), HloOpcode::kRsqrt, var_plus_epsilon)); in HandleBatchNormGrad()
|
| D | elemental_ir_emitter.cc | 254 return EmitDeviceMathCall(TargetDeviceFunctionID::kRsqrt, {value}, in EmitRsqrt()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | identify_l2_normalization_test.cc | 106 EXPECT_FALSE(op->type == OperatorType::kRsqrt); in RunIdentifyL2Normalization()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode.h | 132 V(kRsqrt, "rsqrt", 1) \
|
| D | batchnorm_expander.cc | 90 return HloInstruction::CreateUnary(operand->shape(), HloOpcode::kRsqrt, in Rsqrt()
|
| D | instruction_fusion.cc | 177 case HloOpcode::kRsqrt: in IsExpensive()
|
| D | hlo_instruction.cc | 908 case HloOpcode::kRsqrt: in CreateUnary()
1734 case HloOpcode::kRsqrt: in CloneWithNewOperands()
2161 case HloOpcode::kRsqrt: in IdenticalSlowPath()
2633 case HloOpcode::kRsqrt: in IsOpElementwise()
3184 case HloOpcode::kRsqrt: in Visit()
|
| D | hlo_cost_analysis.cc | 107 opcode == HloOpcode::kRsqrt || opcode == HloOpcode::kTanh || in HandleElementwiseOp()
|
| D | elemental_ir_emitter.cc | 464 case HloOpcode::kRsqrt: in EmitFloatUnaryOp()
795 case HloOpcode::kRsqrt: { in EmitComplexUnaryOp()
2266 case HloOpcode::kRsqrt: in MakeElementGenerator()
|
| D | hlo_graph_dumper.cc | 975 case HloOpcode::kRsqrt: in GetInstructionColor()
|
| D | conditional_code_motion.cc | 194 case HloOpcode::kRsqrt: in WorthHoisting()
|
| D | algebraic_simplifier_test.cc | 1480 HloInstruction::CreateUnary(r0f32, HloOpcode::kRsqrt, param0)); in TEST_F()
6695 EXPECT_EQ(FindInstruction(m.get(), HloOpcode::kRsqrt), nullptr); in TEST_F()
6730 EXPECT_EQ(FindInstruction(m.get(), HloOpcode::kRsqrt), nullptr); in TEST_F()
6765 EXPECT_EQ(FindInstruction(m.get(), HloOpcode::kRsqrt), nullptr); in TEST_F()
6799 EXPECT_NE(FindInstruction(m.get(), HloOpcode::kRsqrt), nullptr); in TEST_F()
|
| D | sharding_propagation.cc | 203 case HloOpcode::kRsqrt: in PickRepresentativeOperand()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | op_version.cc | 251 {{OperatorType::kRsqrt, 1}, "1.10.0"}, in GetMinimumRuntimeVersionForModel()
|
| D | operator_test.cc | 140 CheckSimpleOperator<TensorFlowRsqrtOperator>("RSQRT", OperatorType::kRsqrt); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 126 kRsqrt, enumerator
1367 TensorFlowRsqrtOperator() : Operator(OperatorType::kRsqrt) {}
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 407 return xla::HloOpcode::kRsqrt; in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 781 NoAttributeCase(kRsqrt, RsqrtOp); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 360 case HloOpcode::kRsqrt: in EmitOp()
|