Searched defs:pool_params (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_pooling_ops_common.h | 514 MklPoolParameters* pool_params, in InitMklPoolParameters()
526 void PoolParamsToDims(const MklPoolParameters* pool_params, in PoolParamsToDims()
560 MklPoolParameters* pool_params, in AllocateEmptyOutputTensor()
624 MklPoolParameters* pool_params, in ConfigureInput()
|
| H A D | mkl_avgpooling_op.cc | 61 MklPoolParameters pool_params; in Compute() local
203 MklPoolParameters pool_params; in Compute() local
|
| H A D | mkl_maxpooling_op.cc | 63 MklPoolParameters pool_params; in Compute() local
259 MklPoolParameters pool_params; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| H A D | utils.cc | 237 const TfLitePoolParams* pool_params = in IsNodeSupportedByHexagon() local
250 const TfLitePoolParams* pool_params = in IsNodeSupportedByHexagon() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | xnnpack_delegate.cc | 1897 const TfLitePoolParams* pool_params = in VisitNode() local
1991 const TfLitePoolParams* pool_params = in VisitNode() local
2124 TfLitePoolParams pool_params = {kTfLitePaddingUnknown}; in VisitNode() local
2131 TfLitePoolParams pool_params = {kTfLitePaddingUnknown}; in VisitNode() local
2244 const TfLiteTensor* tensors, const TfLitePoolParams* pool_params, in VisitAveragePool2DNode()
3290 const TfLiteTensor* tensors, const TfLitePoolParams* pool_params, in VisitMaxPool2DNode()
3620 const TfLiteTensor* tensors, const TfLitePoolParams* pool_params, in VisitMediaPipeMaxPoolingNode()
3684 const TfLiteTensor* tensors, const TfLitePoolParams* pool_params, in VisitMediaPipeUnpoolingNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | pool_2d_builder.cc | 38 const TfLitePoolParams* pool_params = in PopulateSubGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | perception_ops_utils.cc | 151 TfLitePoolParams pool_params; in CreateCustomOptions() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | builtin_op_data_test.cc | 31 TfLitePoolParams pool_params; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A D | flatbuffer_export.cc | 454 TfLitePoolParams pool_params; in GetTflitePoolParams() local
|