Searched refs:lrn_op (Results 1 – 4 of 4) 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 | 939 tensorflow::NodeDef* lrn_op = tensorflow_graph->add_node(); in ConvertLocalResponseNormalizationOperator() local
940 lrn_op->set_op("LRN"); in ConvertLocalResponseNormalizationOperator()
941 lrn_op->set_name(src_op.outputs[0]); in ConvertLocalResponseNormalizationOperator()
942 *lrn_op->add_input() = src_op.inputs[0]; in ConvertLocalResponseNormalizationOperator()
943 (*lrn_op->mutable_attr())["depth_radius"].set_i(src_op.range); in ConvertLocalResponseNormalizationOperator()
944 (*lrn_op->mutable_attr())["bias"].set_f(src_op.bias); in ConvertLocalResponseNormalizationOperator()
945 (*lrn_op->mutable_attr())["alpha"].set_f(src_op.alpha); in ConvertLocalResponseNormalizationOperator()
946 (*lrn_op->mutable_attr())["beta"].set_f(src_op.beta); in ConvertLocalResponseNormalizationOperator()
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 150 tensorflow/core/kernels/lrn_op.cc
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3868 ":lrn_op",
3937 name = "lrn_op",
3938 prefix = "lrn_op",
5766 "lrn_op.cc",
|