Home
last modified time | relevance | path

Searched refs:AvgPool (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_AvgPool.pbtxt2 graph_op_name: "AvgPool"
4 name: "nn.AvgPool"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DAvgPool.pbtxt2 name: "AvgPool"
59 name: "AvgPool"
116 name: "AvgPool"
173 name: "AvgPool"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DAvgPool.pbtxt2 name: "AvgPool"
59 name: "AvgPool"
116 name: "AvgPool"
173 name: "AvgPool"
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling_test.cc115 AvgPool(input, kernel_size, stride, padding, data_format, in XLA_TEST_F()
131 AvgPool(input, kernel_size, stride, padding, data_format, in XLA_TEST_F()
147 AvgPool(input, kernel_size, stride, padding, data_format, in XLA_TEST_F()
163 AvgPool(input, kernel_size, stride, {{1, 1}, {2, 1}}, data_format, in XLA_TEST_F()
178 AvgPool(input, kernel_size, stride, {{2, 1}, {1, 1}}, data_format, in XLA_TEST_F()
Dpooling.h57 XlaOp AvgPool(XlaOp operand, absl::Span<const int64> kernel_size,
Dpooling.cc148 XlaOp AvgPool(XlaOp operand, absl::Span<const int64> kernel_size, in AvgPool() function
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_AvgPool.pbtxt2 graph_op_name: "AvgPool"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_AvgPool.pbtxt2 graph_op_name: "AvgPool"
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad_test.cc29 using ops::AvgPool;
282 auto y = AvgPool(scope_, x, ksize, strides, "SAME"); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf-ops.mlir468 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NHWC", ksize = [1, 7, 7, 1], padd…
477 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", ksize = [1, 7, 7, 1], padding = "VALID", strides…
486 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_INT", data_format = "NHWC", ksize = [1, 7, 7, 1], paddin…
495 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "HWCN", ksize = [1, 7, 7, 1], padd…
504 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", padding = "VALID", strides = [1, 1, 1, 1]} : (te…
513 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", ksize = [7, 7, 1], padding = "VALID", strides = …
522 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", ksize = [1, 7, 7.5, 1], padding = "VALID", strid…
530 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", ksize = [1 : i32, 7 : i32, 7 : i32, 1 : i32], pa…
539 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", ksize = [1, 7, 7, 1], strides = [1, 1, 1, 1]} : …
548 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", ksize = [1, 7, 7, 1], padding = "MAGIC", strides…
[all …]
Dlegalize_hlo.mlir1742 // CHECK: %[[VAL_1:.*]] = "tf.AvgPool"(%[[VAL_0]]) {data_format = "NHWC", ksize = [1, 3, …
1764 // CHECK: %[[VAL_1:.*]] = "tf.AvgPool"(%[[VAL_0]]) {data_format = "NHWC", ksize = [1, 3, …
1818 // CHECK: %[[VAL_1:.*]] = "tf.AvgPool"(%[[VAL_0]]) {data_format = "NHWC", ksize = [1, 3, …
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf.mlir100 …%0 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NHWC", ksize = [1, 3, 6, 1], padd…
102 …%1 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NCHW", ksize = [1, 3, 6, 1], padd…
104 …%2 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NHWC", ksize = [3, 3, 6, 1], padd…
106 …%3 = "tf.AvgPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NHWC", ksize = [1, 3, 6, 1], padd…
115 // CHECK: %1 = "tf.AvgPool"(%arg0)
116 // CHECK: %2 = "tf.AvgPool"(%arg0)
117 // CHECK: %3 = "tf.AvgPool"(%arg0)
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc246 xla::AvgPool(converted_input, ksize, stride, spatial_padding, in Compile()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md51 * `raw_ops.AvgPool`
/external/tensorflow/tensorflow/lite/tools/benchmark/
DREADME.md283 … 0.036 0.036 0.001% 1.956% 0.000 0 [MobilenetV1/Logits/AvgPool_1a/AvgPool]
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md27 `AvgPool` | `T={double,float}`
Dcpu_supported_ops.md27 `AvgPool` | `T={double,float}`
/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
Dtf-to-tosa-pipeline.mlir427 …%2 = "tf.AvgPool"(%arg0) {data_format = "NHWC", ksize = [1, 1, 1, 1], padding = "SAME", strides =…
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc459 Output avg_pool_op = AvgPool(root.WithOpName("avg_pool_op"), constant_op, in TestQuantizeAvgPool()
/external/llvm-project/mlir/docs/
DOpDefinitions.md103 for all the fields it requires. For example, `tf.AvgPool` is defined as
106 def TF_AvgPoolOp : TF_Op<"AvgPool", [NoSideEffect]> {
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md1305 ### tf.AvgPool
1312 %output = tf.AvgPool(%value) {ksize, strides, padding, data_format}
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc81 DECL_CONVERT_OP(AvgPool);
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir4398 // tf.AvgPool legalization
4421 …%0 = "tf.AvgPool"(%arg0) {data_format = "NHWC", ksize = [1, 2, 2, 1], padding = "VALID", strides =…
4469 …%0 = "tf.AvgPool"(%arg0) {data_format = "NCHW", ksize = [1, 1, 2, 2], padding = "VALID", strides =…
4523 …%0 = "tf.AvgPool"(%arg0) {data_format = "NHWC", ksize = [1, 5, 2, 1], padding = "SAME", strides = …
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1107 TEST_F(OpTest, AvgPool) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc5067 return CreatePoolOp<ops::AvgPool>(tf_type, ksize, strides, padding, in TEST_P()

12