Searched refs:rsqrt_op (Results 1 – 4 of 4) sorted by relevance
77 auto rsqrt_op = new TensorFlowRsqrtOperator; in RunIdentifyL2Normalization() local78 rsqrt_op->inputs = {sum_op->outputs[0]}; in RunIdentifyL2Normalization()79 rsqrt_op->outputs = {"Rsquareoutput"}; in RunIdentifyL2Normalization()85 mul_op->inputs = {"input0", rsqrt_op->outputs[0]}; in RunIdentifyL2Normalization()90 model.operators.push_back(std::unique_ptr<Operator>(rsqrt_op)); in RunIdentifyL2Normalization()
1805 auto* rsqrt_op = new TensorFlowRsqrtOperator; in ConvertBatchNormWithGlobalNormalizationOperator() local1806 rsqrt_op->inputs.push_back(node.input(2)); in ConvertBatchNormWithGlobalNormalizationOperator()1807 rsqrt_op->outputs.push_back(rsqrt); in ConvertBatchNormWithGlobalNormalizationOperator()1808 model->operators.emplace_back(rsqrt_op); in ConvertBatchNormWithGlobalNormalizationOperator()1818 auto* rsqrt_op = new TensorFlowRsqrtOperator; in ConvertBatchNormWithGlobalNormalizationOperator() local1819 rsqrt_op->inputs.push_back(node.input(2)); in ConvertBatchNormWithGlobalNormalizationOperator()1820 rsqrt_op->outputs.push_back(multiplier); in ConvertBatchNormWithGlobalNormalizationOperator()1821 model->operators.emplace_back(rsqrt_op); in ConvertBatchNormWithGlobalNormalizationOperator()1865 auto* rsqrt_op = new TensorFlowRsqrtOperator; in ConvertFusedBatchNormOperator() local1866 rsqrt_op->inputs.push_back(epsilon_add_op_name); in ConvertFusedBatchNormOperator()[all …]
930 tensorflow::NodeDef* rsqrt_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local931 rsqrt_op->set_op("Rsqrt"); in ConvertL2NormalizationOperator()932 rsqrt_op->set_name(rsqrt_output); in ConvertL2NormalizationOperator()933 *rsqrt_op->add_input() = sum_output; in ConvertL2NormalizationOperator()934 (*rsqrt_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertL2NormalizationOperator()1137 tensorflow::NodeDef* rsqrt_op = tensorflow_graph->add_node(); in ConvertRsqrtOperator() local1138 rsqrt_op->set_op("Rsqrt"); in ConvertRsqrtOperator()1139 rsqrt_op->set_name(src_op.outputs[0]); in ConvertRsqrtOperator()1141 *rsqrt_op->add_input() = src_op.inputs[0]; in ConvertRsqrtOperator()1144 (*rsqrt_op->mutable_attr())["T"].set_type(data_type); in ConvertRsqrtOperator()
1226 ::mlir::TF::RsqrtOp rsqrt_op; in matchAndRewrite()1231 rsqrt_op = rewriter.create<::mlir::TF::RsqrtOp>(odsLoc, tblgen_values, in matchAndRewrite()1237 ::mlir::Value tblgen_value_1 = (*rsqrt_op.getODSResults(0).begin()); in matchAndRewrite()