Home
last modified time | relevance | path

Searched refs:avg_pool (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py201 avg_pool_func = nn_ops.avg_pool
245 nn_ops.avg_pool,
256 nn_ops.avg_pool,
273 nn_ops.avg_pool,
286 nn_ops.avg_pool,
296 nn_ops.avg_pool,
307 nn_ops.avg_pool,
320 nn_ops.avg_pool,
328 nn_ops.avg_pool,
342 nn_ops.avg_pool,
[all …]
Dfractional_avg_pool_op_test.py367 output_tensor = nn_ops.avg_pool(input_tensor, window_size,
406 output_tensor = nn_ops.avg_pool(input_tensor, window_size,
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_AvgPoolGrad.pbtxt7 1-D. Shape of the original input to `avg_pool`.
14 the output of `avg_pool`.
20 4-D. Gradients w.r.t. the input of `avg_pool`.
/external/tensorflow/tensorflow/compiler/tests/
Dpooling_ops_test.py273 nn_ops.avg_pool,
283 nn_ops.avg_pool,
565 self._TestPooling(nn_ops.avg_pool, AvgPoolGrad)
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dops.py57 x = tf.nn.avg_pool(
/external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/
Ddensenet.py105 self.avg_pool = tf.keras.layers.AveragePooling2D(data_format=data_format)
110 output = self.avg_pool(output)
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt17 avg_pool/avg_pool_data_format='NHWC',input_shape=[1,1,1,1],ksize=[1,1,1,1],padding='SAME',strides=[…
18 avg_pool/avg_pool_data_format='NHWC',input_shape=[1,1,1,1],ksize=[1,1,1,1],padding='SAME',strides=[…
19 avg_pool/avg_pool_data_format='NHWC',input_shape=[1,1,1,1],ksize=[1,1,1,1],padding='SAME',strides=[…
20 avg_pool/avg_pool_data_format='NHWC',input_shape=[1,15,14,1],ksize=[1,1,1,1],padding='SAME',strides…
21 avg_pool/avg_pool_data_format='NHWC',input_shape=[1,15,14,1],ksize=[1,1,1,1],padding='SAME',strides…
22 avg_pool/avg_pool_data_format='NHWC',input_shape=[1,15,14,1],ksize=[1,1,1,1],padding='SAME',strides…
23 avg_pool/avg_pool_data_format='NHWC',input_shape=[3,15,14,3],ksize=[1,1,1,1],padding='SAME',strides…
24 avg_pool/avg_pool_data_format='NHWC',input_shape=[3,15,14,3],ksize=[1,1,1,1],padding='SAME',strides…
25 avg_pool/avg_pool_data_format='NHWC',input_shape=[3,15,14,3],ksize=[1,1,1,1],padding='SAME',strides…
26 avg_pool/avg_pool_data_format='NHWC',input_shape=[1,1,1,1],ksize=[1,1,1,1],padding='VALID',strides=…
[all …]
/external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/
Dresnet50.py255 self.avg_pool = layers.AveragePooling2D(
301 x = self.avg_pool(x)
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py1201 ("AVG", 1): avg_pool,
1202 ("AVG", 2): avg_pool,
3415 2: gen_nn_ops.avg_pool,
3430 def avg_pool(value, ksize, strides, padding, data_format="NHWC", function
3464 return gen_nn_ops.avg_pool(
3503 return gen_nn_ops.avg_pool(
3547 result = gen_nn_ops.avg_pool(
Dnn_test.py1273 y2 = nn_ops.avg_pool(x, ksize, strides, "SAME")
1283 y2 = nn_ops.avg_pool(x, ksize, strides, "SAME")
Dnn_grad.py665 gen_nn_ops.avg_pool(
Dimage_ops_impl.py3087 downscaled = [nn_ops.avg_pool(x, ksize=divisor, strides=divisor,
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_3d_sycl.h613 AvgPool3DSYCL<T> avg_pool(depth, batch, in_planes, in_rows, in_cols,
617 cgh.parallel_for(cl::sycl::range<1>(num_threads), avg_pool);
/external/tensorflow/tensorflow/python/keras/layers/
Dpooling.py376 nn.avg_pool,
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_compatibility.md50 * [tf.nn.avg_pool](https://www.tensorflow.org/api_docs/python/tf/nn/avg_pool)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt32 name: "avg_pool"
/external/tensorflow/tensorflow/lite/
Dbuild_def.bzl234 "avg_pool",
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc244 csinfo_.avg_pool = "AvgPool"; in MklLayoutRewritePass()
342 rinfo_.push_back({csinfo_.avg_pool, in MklLayoutRewritePass()
343 mkl_op_registry::GetMklOpName(csinfo_.avg_pool), in MklLayoutRewritePass()
678 string avg_pool; member
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt24 name: "avg_pool"
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops_test.py290 output = nn.avg_pool(
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples.py655 make_pool_tests(tf.nn.avg_pool)(zip_path, expected_tf_failures=80)
3077 return tf.sqrt(tf.nn.avg_pool(
/external/tensorflow/tensorflow/python/keras/
Dbackend.py4701 x = nn.avg_pool(