| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | test_manifest.txt | 1183 local_response_norm/local_response_norm_alpha=None,beta=None,bias=None,depth_radius=None,input_shap…
1184 local_response_norm/local_response_norm_alpha=None,beta=None,bias=None,depth_radius=0,input_shape=[…
1185 local_response_norm/local_response_norm_alpha=None,beta=None,bias=None,depth_radius=1,input_shape=[…
1186 local_response_norm/local_response_norm_alpha=None,beta=None,bias=None,depth_radius=3,input_shape=[…
1187 local_response_norm/local_response_norm_alpha=None,beta=None,bias=None,depth_radius=5,input_shape=[…
1188 local_response_norm/local_response_norm_alpha=None,beta=0.25,bias=None,depth_radius=None,input_shap…
1189 local_response_norm/local_response_norm_alpha=None,beta=0.25,bias=None,depth_radius=0,input_shape=[…
1190 local_response_norm/local_response_norm_alpha=None,beta=0.25,bias=None,depth_radius=1,input_shape=[…
1191 local_response_norm/local_response_norm_alpha=None,beta=0.25,bias=None,depth_radius=3,input_shape=[…
1192 local_response_norm/local_response_norm_alpha=None,beta=0.25,bias=None,depth_radius=5,input_shape=[…
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_LRN.pbtxt | 10 name: "depth_radius"
38 `depth_radius`. In detail,
41 sum(input[a, b, c, d - depth_radius : d + depth_radius + 1] ** 2)
|
| D | api_def_LRNGrad.pbtxt | 29 name: "depth_radius"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lrn_op_test.cc | 63 const int64 depth_radius = GetIntAttr("depth_radius"); in Compare() local
77 for (int64 r = std::max(int64{0}, d - depth_radius); in Compare()
78 r < std::min(depth, d + depth_radius + 1); ++r) { in Compare()
203 int depth_radius) { in MakeRNGrad() argument
218 .Attr("depth_radius", depth_radius) in MakeRNGrad()
|
| D | lrn_op.cc | 63 void GetBandMatrix(int depth, int depth_radius, in GetBandMatrix() argument
67 const int begin = std::max<int>(0, row - depth_radius); in GetBandMatrix()
68 const int end = std::min<int>(depth, row + depth_radius + 1); in GetBandMatrix()
85 LaunchLRN(int depth_radius, T bias, T alpha, T beta) in LaunchLRN()
86 : depth_radius_(depth_radius), bias_(bias), alpha_(alpha), beta_(beta) {} in LaunchLRN()
179 LaunchLRN(int depth_radius, T bias, T alpha, T beta) in LaunchLRN()
180 : depth_radius_(depth_radius), bias_(bias), alpha_(alpha), beta_(beta) {} in LaunchLRN()
388 LaunchLRNGrad(int depth_radius, T bias, T alpha, T beta) in LaunchLRNGrad()
389 : depth_radius_(depth_radius), in LaunchLRNGrad()
471 LaunchLRNGrad(int depth_radius, T bias, T alpha, T beta) in LaunchLRNGrad()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | lrn_ops_test.py | 76 depth_radius=lrn_depth_radius,
108 depth_radius = np.random.randint(1, shape[3])
119 depth_radius, bias, alpha, beta)
122 depth_radius, bias, alpha, beta)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | LRN.pbtxt | 12 name: "depth_radius"
64 name: "depth_radius"
|
| D | LRNGrad.pbtxt | 20 name: "depth_radius"
80 name: "depth_radius"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | LRN.pbtxt | 12 name: "depth_radius"
64 name: "depth_radius"
|
| D | LRNGrad.pbtxt | 20 name: "depth_radius"
80 name: "depth_radius"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | lrn_op_test.py | 75 depth_radius=lrn_depth_radius,
141 depth_radius=lrn_depth_radius,
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | local_response_norm.py | 45 depth_radius=parameters["depth_radius"],
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 54 void GetBandMatrix(int depth, int depth_radius, in GetBandMatrix() argument
58 const int begin = std::max<int>(0, row - depth_radius); in GetBandMatrix()
59 const int end = std::min<int>(depth, row + depth_radius + 1); in GetBandMatrix()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 201 …argspec: "args=[\'input\', \'depth_radius\', \'bias\', \'alpha\', \'beta\', \'name\'], varargs=Non…
213 …argspec: "args=[\'input\', \'depth_radius\', \'bias\', \'alpha\', \'beta\', \'name\'], varargs=Non…
|
| D | tensorflow.raw_ops.pbtxt | 2073 …argspec: "args=[\'input\', \'depth_radius\', \'bias\', \'alpha\', \'beta\', \'name\'], varargs=Non…
2077 …argspec: "args=[\'input_grads\', \'input_image\', \'output_image\', \'depth_radius\', \'bias\', \'…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 650 depth_radius = op.get_attr("depth_radius")
655 gen_nn_ops.lrn_grad(grad, op.inputs[0], op.outputs[0], depth_radius, bias,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 237 …argspec: "args=[\'input\', \'depth_radius\', \'bias\', \'alpha\', \'beta\', \'name\'], varargs=Non…
253 …argspec: "args=[\'input\', \'depth_radius\', \'bias\', \'alpha\', \'beta\', \'name\'], varargs=Non…
|
| D | tensorflow.raw_ops.pbtxt | 2073 …argspec: "args=[\'input\', \'depth_radius\', \'bias\', \'alpha\', \'beta\', \'name\'], varargs=Non…
2077 …argspec: "args=[\'input_grads\', \'input_image\', \'output_image\', \'depth_radius\', \'bias\', \'…
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1595 int depth_radius; in LrnRewrite() local
1596 TF_CHECK_OK(GetNodeAttr(n->def(), "depth_radius", &depth_radius)); in LrnRewrite()
1600 if (depth_radius == 2) { in LrnRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 1155 `depth_radius`. In detail,
1158 sum(input[a, b, c, d - depth_radius : d + depth_radius + 1] ** 2)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf.mlir | 1372 …%0 = "tf.LRN"(%arg0) {depth_radius = 5 :i64, bias = 1.0 :f32, alpha = 1.0 : f32, beta = 0.5 :f32} …
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 14832 name: "depth_radius"
14871 name: "depth_radius"
14931 name: "depth_radius"
14978 name: "depth_radius"
|
| D | ops_history.v2.pbtxt | 30896 name: "depth_radius"
30948 name: "depth_radius"
31009 name: "depth_radius"
31069 name: "depth_radius"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 5970 `depth_radius`. In detail,
5973 sum(input[a, b, c, d - depth_radius : d + depth_radius + 1] ** 2)
5983 DefaultValuedAttr<I64Attr, "5">:$depth_radius,
6004 DefaultValuedAttr<I64Attr, "5">:$depth_radius,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 20607 name: "depth_radius"
20668 name: "depth_radius"
|