Searched refs:lrn_op (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | lrn_op_test.py | 110 lrn_op = nn.local_response_normalization(p, 2, 1.0, 0.0, 1.0, name="lrn")
111 grad = gradients_impl.gradients([lrn_op], [p])[0]
138 lrn_op = nn.local_response_normalization(
145 err = gradient_checker.compute_gradient_error(inp, shape, lrn_op, shape)
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 947 tensorflow::NodeDef* lrn_op = tensorflow_graph->add_node(); in ConvertLocalResponseNormalizationOperator() local
948 lrn_op->set_op("LRN"); in ConvertLocalResponseNormalizationOperator()
949 lrn_op->set_name(src_op.outputs[0]); in ConvertLocalResponseNormalizationOperator()
950 *lrn_op->add_input() = src_op.inputs[0]; in ConvertLocalResponseNormalizationOperator()
951 (*lrn_op->mutable_attr())["depth_radius"].set_i(src_op.range); in ConvertLocalResponseNormalizationOperator()
952 (*lrn_op->mutable_attr())["bias"].set_f(src_op.bias); in ConvertLocalResponseNormalizationOperator()
953 (*lrn_op->mutable_attr())["alpha"].set_f(src_op.alpha); in ConvertLocalResponseNormalizationOperator()
954 (*lrn_op->mutable_attr())["beta"].set_f(src_op.beta); in ConvertLocalResponseNormalizationOperator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3890 ":lrn_op",
3968 name = "lrn_op",
3969 prefix = "lrn_op",
6152 "lrn_op.cc",
|