Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DMaxPool.pbtxt2 name: "MaxPool"
61 name: "MaxPool"
127 name: "MaxPool"
195 name: "MaxPool"
264 name: "MaxPool"
/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/compiler/mlir/tensorflow/tests/
Dlayout_optimization_move_transposes_end.mlir55 // MaxPool operand transpose must be folded into the op and MaxPool
59 …// CHECK: %[[MAX_POOL:[0-9]*]] = "tf.MaxPool"(%arg0) {data_format = "NCHW", ksize = [1, 1, 3, 3], …
67 // Compute MaxPool in NHWC format
68 %2 = "tf.MaxPool"(%1)
Dlayout_optimization_layout_assignment_to_nhwc.mlir90 …// CHECK: %[[R1:.*]] = "tf.MaxPool"(%[[R0]]) {data_format = "NHWC", explicit_paddings = [], ksize …
93 %0 = "tf.MaxPool"(%arg0)
Dlayout_optimization_to_nhwc.mlir73 %8 = "tf.MaxPool"(%7)
81 // CHECK: %[[MAX_POOL:[0-9]*]] = "tf.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,
/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.cc121 reference_ops::MaxPool(op_params, tflite::micro::GetTensorShape(input), in MaxEvalFloat()
141 reference_ops::MaxPool(op_params, tflite::micro::GetTensorShape(input), in MaxEvalQuantized()
146 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
125 name: "MaxPool"
126 op: "MaxPool"
/external/tensorflow/tensorflow/lite/kernels/
Dpooling.cc246 type::MaxPool(op_params, GetTensorShape(input), GetTensorData<float>(input), \ in MaxEvalFloat()
274 type::MaxPool(op_params, GetTensorShape(input), \ in MaxEvalQuantizedUInt8()
303 type::MaxPool(op_params, GetTensorShape(input), \ in MaxEvalQuantizedInt8()
332 type::MaxPool(op_params, GetTensorShape(input), \ in MaxEvalQuantizedInt16()
Dpooling_test.cc621 TEST(FloatPoolingOpTest, MaxPool) { in TEST() argument
718 TEST(QuantizedUInt8PoolingOpTest, MaxPool) { in TEST() argument
887 TEST(QuantizedInt8PoolingOpTest, MaxPool) { in TEST() argument
907 TEST(QuantizedInt8PoolingOpTest16, MaxPool) { in TEST() argument
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_MaxPool.pbtxt2 graph_op_name: "MaxPool"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_op_interfaces.td75 Example: MaxPool(Transpose($arg, $perm))
76 -> Transpose(MaxPool($arg, $perm))
/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_pool_with_argmax.cc33 inline void MaxPool(const PoolParams& params, const RuntimeShape& input_shape, in MaxPool() function
227 MaxPool<float>(op_params, GetTensorShape(input), GetTensorShape(output), in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dpooling.cc286 reference_ops::MaxPool(op_params, tflite::micro::GetTensorShape(input), in MaxEvalFloat()
313 reference_ops::MaxPool(op_params, tflite::micro::GetTensorShape(input), in MaxEvalQuantized()
318 reference_integer_ops::MaxPool( in MaxEvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dpooling.h82 inline void MaxPool(const PoolParams& params, const RuntimeShape& input_shape, in MaxPool() function
198 inline void MaxPool(const PoolParams& params, const RuntimeShape& input_shape, in MaxPool() function
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dpooling.h190 inline void MaxPool(const PoolParams& params, const RuntimeShape& input_shape, in MaxPool() function
240 inline void MaxPool(const PoolParams& params, const RuntimeShape& input_shape, in MaxPool() function
Dlegacy_reference_ops.h1638 inline void MaxPool(const float* input_data, const Dims<4>& input_dims, in MaxPool() function
1652 MaxPool(params, DimsToShape(input_dims), input_data, DimsToShape(output_dims), in MaxPool()
1658 void MaxPool(const float* input_data, const Dims<4>& input_dims, in MaxPool() function
1664 MaxPool(input_data, input_dims, stride_width, stride_height, pad_width, in MaxPool()
1671 void MaxPool(const float* input_data, const Dims<4>& input_dims, int stride, in MaxPool() function
1674 MaxPool<Ac>(input_data, input_dims, stride, stride, pad_width, pad_height, in MaxPool()
1678 inline void MaxPool(const uint8* input_data, const Dims<4>& input_dims, in MaxPool() function
1692 MaxPool(params, DimsToShape(input_dims), input_data, DimsToShape(output_dims), in MaxPool()
1698 void MaxPool(const uint8* input_data, const Dims<4>& input_dims, in MaxPool() function
1712 MaxPool(input_data, input_dims, stride_width, stride_height, pad_width, in MaxPool()
[all …]
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad_test.cc39 using ops::MaxPool;
243 auto y = MaxPool(scope_, x, ksize, strides, "VALID"); in TEST_F()

123