Home
last modified time | relevance | path

Searched refs:rsqrt_op (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc1684 auto* rsqrt_op = new TensorFlowRsqrtOperator; in ConvertBatchNormWithGlobalNormalizationOperator() local
1685 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() local
1698 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() local
1743 rsqrt_op->inputs.push_back(epsilon_add_op_name); in ConvertFusedBatchNormOperator()
[all …]
Dexport_tensorflow.cc922 tensorflow::NodeDef* rsqrt_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local
923 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() local
1130 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()
/external/tensorflow/tensorflow/contrib/quantize/python/
Dfold_batch_norms.py992 rsqrt_op = graph.get_operation_by_name(bn + 'BatchNorm/batchnorm_1/Rsqrt')
993 rsqrt_consumers = input_to_ops_map.ConsumerOperations(rsqrt_op)