| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | pooling_ops_test.py | 300 avg_pool_func = nn_ops.avg_pool
358 @parameterized.parameters(GetTestConfigsDicts(nn_ops.avg_pool))
369 @parameterized.parameters(GetTestConfigsDicts(nn_ops.avg_pool))
380 @parameterized.parameters(GetTestConfigsDicts(nn_ops.avg_pool))
391 @parameterized.parameters(GetTestConfigsDicts(nn_ops.avg_pool))
409 @parameterized.parameters(GetTestConfigsDicts(nn_ops.avg_pool))
424 @parameterized.parameters(GetTestConfigsDicts(nn_ops.avg_pool))
438 @parameterized.parameters(GetTestConfigsDicts(nn_ops.avg_pool))
452 @parameterized.parameters(GetTestConfigsDicts(nn_ops.avg_pool))
463 @parameterized.parameters(GetTestConfigsDicts(nn_ops.avg_pool))
[all …]
|
| D | fractional_avg_pool_op_test.py | 390 output_tensor = nn_ops.avg_pool(input_tensor, window_size,
429 output_tensor = nn_ops.avg_pool(input_tensor, window_size,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_AvgPoolGrad.pbtxt | 7 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/lite/testing/op_tests/ |
| D | pool.py | 118 tf.nn.avg_pool(
134 tf.nn.avg_pool, allow_fully_quantize=True)(
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_test.py | 269 nn_ops.avg_pool,
279 nn_ops.avg_pool,
561 self._TestPooling(nn_ops.avg_pool, AvgPoolGrad)
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | test_manifest.txt | 17 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/python/eager/benchmarks/resnet50/ |
| D | resnet50.py | 272 self.avg_pool = layers.AveragePooling2D((7, 7),
353 x = self.avg_pool(x)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 639 auto avg_pool = rewriter.create<TFL::AveragePool2DOp>( in matchAndRewrite() local
645 auto value_to_replace = avg_pool.getResult(); in matchAndRewrite()
653 InsertReshapeOp(mean_op.getLoc(), avg_pool.getResult(), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 1628 ("AVG", 1): avg_pool,
1629 ("AVG", 2): avg_pool,
4518 2: gen_nn_ops.avg_pool,
4534 def avg_pool(value, ksize, strides, padding, data_format="NHWC", function
4568 return gen_nn_ops.avg_pool(
4609 return gen_nn_ops.avg_pool(
4655 result = gen_nn_ops.avg_pool(
|
| D | nn_grad.py | 668 gen_nn_ops.avg_pool(
|
| D | nn_test.py | 1610 y2 = nn_ops.avg_pool(x, ksize, strides, "SAME")
1623 y2 = nn_ops.avg_pool(x, ksize, strides, "SAME")
|
| D | image_ops_impl.py | 4547 nn_ops.avg_pool(
|
| /external/android-nn-driver/docs/ |
| D | FAQ.md | 38 * Some TestRandomGraph/RandomGraphTest tests which include avg_pool or instance_normalization opera…
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | pooling.py | 642 nn.avg_pool,
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | BUILD | 644 "avg_pool",
|
| D | build_def.bzl | 20 "avg_pool",
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 40 name: "avg_pool"
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 245 csinfo_.avg_pool = "AvgPool"; in MklLayoutRewritePass()
394 rinfo_.push_back({csinfo_.avg_pool, in MklLayoutRewritePass()
395 mkl_op_registry::GetMklOpName(csinfo_.avg_pool), in MklLayoutRewritePass()
855 string avg_pool; member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 3764 auto avg_pool = llvm::dyn_cast<AveragePool2DOp>(op); in GetArithmeticCount() local
3765 return avg_pool.filter_height() * avg_pool.filter_width() * count; in GetArithmeticCount()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 32 name: "avg_pool"
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 621 output = nn.avg_pool(
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 5755 x = nn.avg_pool(
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 998 Arg<TF_Int32Tensor, [{1-D. Shape of the original input to `avg_pool`.}]>:$orig_input_shape,
1000 the output of `avg_pool`.}]>:$grad,
1009 Res<TF_FloatTensor, [{4-D. Gradients w.r.t. the input of `avg_pool`.}]>:$output
|