Home
last modified time | relevance | path

Searched refs:MaxPool (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DMaxPool.pbtxt2 name: "MaxPool"
61 name: "MaxPool"
127 name: "MaxPool"
195 name: "MaxPool"
/external/tensorflow/tensorflow/lite/kernels/internal/
Dmaxpool_quantized_test.cc42 reference_integer_ops::MaxPool(params, input_shape, input_data, output_shape, in RunOneMaxPoolTest()
44 optimized_integer_ops::MaxPool(params, input_shape, input_data, output_shape, in RunOneMaxPoolTest()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling_test.cc73 MaxPool(input, kernel_size, stride, Padding::kValid, data_format); in XLA_TEST_F()
86 MaxPool(input, kernel_size, stride, Padding::kSame, data_format); in XLA_TEST_F()
99 MaxPool(input, kernel_size, stride, Padding::kSame, data_format); in XLA_TEST_F()
Dpooling.h52 XlaOp MaxPool(XlaOp operand, absl::Span<const int64> kernel_size,
Dpooling.cc133 XlaOp MaxPool(XlaOp operand, absl::Span<const int64> kernel_size, in MaxPool() function
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_MaxPool.pbtxt2 graph_op_name: "MaxPool"
Dapi_def_MaxPoolV2.pbtxt4 name: "nn.MaxPool"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_MaxPool.pbtxt2 graph_op_name: "MaxPool"
/external/tensorflow/tensorflow/lite/micro/kernels/
Dpooling.cc123 reference_ops::MaxPool(op_params, GetTensorShape(input), in MaxEvalFloat()
148 reference_ops::MaxPool( in MaxEvalQuantized()
152 reference_integer_ops::MaxPool( in MaxEvalQuantized()
/external/tensorflow/tensorflow/compiler/jit/tests/
Dkeras_imagenet_main_graph_mode.golden_summary33 MaxPool 1
Dkeras_imagenet_main.golden_summary32 MaxPool 1
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dgraph-default-attr.pbtxt10 # CHECK: tf.MaxPool
124 name: "MaxPool"
125 op: "MaxPool"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_MaxPool.pbtxt2 graph_op_name: "MaxPool"
/external/tensorflow/tensorflow/lite/kernels/
Dpooling.cc217 type::MaxPool(op_params, GetTensorShape(input), GetTensorData<float>(input), \ in MaxEvalFloat()
245 type::MaxPool(op_params, GetTensorShape(input), \ in MaxEvalQuantizedUInt8()
274 type::MaxPool(op_params, GetTensorShape(input), \ in MaxEvalQuantizedInt8()
Dpooling_test.cc575 TEST(FloatPoolingOpTest, MaxPool) { in TEST() argument
672 TEST(QuantizedUInt8PoolingOpTest, MaxPool) { in TEST() argument
841 TEST(QuantizedInt8PoolingOpTest, MaxPool) { in TEST() argument
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis-nn/
Dpooling.cc181 reference_ops::MaxPool(op_params, GetTensorShape(input), in MaxEvalFloat()
202 reference_ops::MaxPool(op_params, GetTensorShape(input), in MaxEvalQuantizedUInt8()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dpooling.h189 inline void MaxPool(const PoolParams& params, const RuntimeShape& input_shape, in MaxPool() function
239 inline void MaxPool(const PoolParams& params, const RuntimeShape& input_shape, in MaxPool() function
Dlegacy_reference_ops.h1668 inline void MaxPool(const float* input_data, const Dims<4>& input_dims, in MaxPool() function
1682 MaxPool(params, DimsToShape(input_dims), input_data, DimsToShape(output_dims), in MaxPool()
1688 void MaxPool(const float* input_data, const Dims<4>& input_dims, in MaxPool() function
1694 MaxPool(input_data, input_dims, stride_width, stride_height, pad_width, in MaxPool()
1701 void MaxPool(const float* input_data, const Dims<4>& input_dims, int stride, in MaxPool() function
1704 MaxPool<Ac>(input_data, input_dims, stride, stride, pad_width, pad_height, in MaxPool()
1708 inline void MaxPool(const uint8* input_data, const Dims<4>& input_dims, in MaxPool() function
1722 MaxPool(params, DimsToShape(input_dims), input_data, DimsToShape(output_dims), in MaxPool()
1728 void MaxPool(const uint8* input_data, const Dims<4>& input_dims, in MaxPool() function
1742 MaxPool(input_data, input_dims, stride_width, stride_height, pad_width, in MaxPool()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/arc/
Dpooling.cc184 reference_ops::MaxPool(op_params, GetTensorShape(input), in MaxEvalFloat()
205 reference_ops::MaxPool(op_params, GetTensorShape(input), in MaxEvalQuantizedUInt8()
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad_test.cc39 using ops::MaxPool;
243 auto y = MaxPool(scope_, x, ksize, strides, "VALID"); in TEST_F()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dpooling.h81 inline void MaxPool(const PoolParams& params, const RuntimeShape& input_shape, in MaxPool() function
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc178 xla::MaxPool(ctx->Input(0), ksize, stride, padding_, in Compile()
335 xla::MaxPool(ctx->Input(0), ksize_, stride_, xla_padding, in Compile()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dpooling.h45 inline void MaxPool(const PoolParams& params, const RuntimeShape& input_shape, in MaxPool() function
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf.mlir278 …%0 = "tf.MaxPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NHWC", ksize = [1, 3, 6, 1], padd…
281 …%1 = "tf.MaxPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NCHW", ksize = [1, 3, 6, 1], padd…
284 …%2 = "tf.MaxPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NHWC", ksize = [3, 3, 6, 1], padd…
287 …%3 = "tf.MaxPool"(%arg0) {T = "tfdtype$DT_FLOAT", data_format = "NHWC", ksize = [1, 3, 6, 1], padd…
296 // CHECK: %1 = "tf.MaxPool"(%arg0)
297 // CHECK: %2 = "tf.MaxPool"(%arg0)
298 // CHECK: %3 = "tf.MaxPool"(%arg0)
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3847 inline void MaxPool(const float* input_data, const Dims<4>& input_dims, in MaxPool() function
3861 MaxPool(params, DimsToShape(input_dims), input_data, DimsToShape(output_dims), in MaxPool()
3867 void MaxPool(const float* input_data, const Dims<4>& input_dims, in MaxPool() function
3873 MaxPool(input_data, input_dims, stride_width, stride_height, pad_width, in MaxPool()
3880 void MaxPool(const float* input_data, const Dims<4>& input_dims, int stride, in MaxPool() function
3883 MaxPool<Ac>(input_data, input_dims, stride, stride, pad_width, pad_height, in MaxPool()
3887 inline void MaxPool(const uint8* input_data, const Dims<4>& input_dims, in MaxPool() function
3901 MaxPool(params, DimsToShape(input_dims), input_data, DimsToShape(output_dims), in MaxPool()
3907 void MaxPool(const uint8* input_data, const Dims<4>& input_dims, in MaxPool() function
3921 MaxPool(input_data, input_dims, stride_width, stride_height, pad_width, in MaxPool()
[all …]

12