Searched refs:rsqrt_op (Results 1 – 3 of 3) sorted by relevance
1684 auto* rsqrt_op = new TensorFlowRsqrtOperator; in ConvertBatchNormWithGlobalNormalizationOperator() local1685 rsqrt_op->inputs.push_back(node.input(2)); in ConvertBatchNormWithGlobalNormalizationOperator()1686 rsqrt_op->outputs.push_back(rsqrt); in ConvertBatchNormWithGlobalNormalizationOperator()1687 model->operators.emplace_back(rsqrt_op); in ConvertBatchNormWithGlobalNormalizationOperator()1697 auto* rsqrt_op = new TensorFlowRsqrtOperator; in ConvertBatchNormWithGlobalNormalizationOperator() local1698 rsqrt_op->inputs.push_back(node.input(2)); in ConvertBatchNormWithGlobalNormalizationOperator()1699 rsqrt_op->outputs.push_back(multiplier); in ConvertBatchNormWithGlobalNormalizationOperator()1700 model->operators.emplace_back(rsqrt_op); in ConvertBatchNormWithGlobalNormalizationOperator()1742 auto* rsqrt_op = new TensorFlowRsqrtOperator; in ConvertFusedBatchNormOperator() local1743 rsqrt_op->inputs.push_back(epsilon_add_op_name); in ConvertFusedBatchNormOperator()[all …]
922 tensorflow::NodeDef* rsqrt_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local923 rsqrt_op->set_op("Rsqrt"); in ConvertL2NormalizationOperator()924 rsqrt_op->set_name(rsqrt_output); in ConvertL2NormalizationOperator()925 *rsqrt_op->add_input() = sum_output; in ConvertL2NormalizationOperator()926 (*rsqrt_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertL2NormalizationOperator()1129 tensorflow::NodeDef* rsqrt_op = tensorflow_graph->add_node(); in ConvertRsqrtOperator() local1130 rsqrt_op->set_op("Rsqrt"); in ConvertRsqrtOperator()1131 rsqrt_op->set_name(src_op.outputs[0]); in ConvertRsqrtOperator()1133 *rsqrt_op->add_input() = src_op.inputs[0]; in ConvertRsqrtOperator()1136 (*rsqrt_op->mutable_attr())["T"].set_type(data_type); in ConvertRsqrtOperator()
992 rsqrt_op = graph.get_operation_by_name(bn + 'BatchNorm/batchnorm_1/Rsqrt')993 rsqrt_consumers = input_to_ops_map.ConsumerOperations(rsqrt_op)