Searched refs:POOLING_2D (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | pooling_test.cc | 61 SingleOpModel model({ToString(OperationType::POOLING_2D), attr}, {input}, in TEST()
90 SingleOpModel model({ToString(OperationType::POOLING_2D), attr}, {input}, in TEST()
116 SingleOpModel model({ToString(OperationType::POOLING_2D), attr}, {input}, in TEST()
|
| D | registry.cc | 88 insert_op(Type::POOLING_2D, NewPoolingNodeShader); in Registry()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | global_pooling_to_reduce_op_test.cc | 49 pool_node->operation.type = ToString(OperationType::POOLING_2D); in TEST()
|
| D | global_pooling_to_reduce_op.cc | 52 if (node->operation.type != ToString(OperationType::POOLING_2D)) { in ApplyToNode()
|
| D | merge_padding_with.cc | 97 OperationType::POOLING_2D); in NewMergePaddingWithPooling()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | operations.h | 69 POOLING_2D, enumerator
|
| D | operations.cc | 143 case OperationType::POOLING_2D: in ToString()
234 {"pooling_2d", OperationType::POOLING_2D}, in OperationTypeFromString()
|
| D | model_builder.cc | 1353 node->operation.type = ToString(OperationType::POOLING_2D); in Parse()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | operation_selector.cc | 447 case OperationType::POOLING_2D: { in GPUOperationFromNode()
|