Searched refs:Pooling2DAttributes (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | global_pooling_to_reduce_op.cc | 34 bool IsGlobalPooling(const Pooling2DAttributes& attr, const BHWC& src_shape, in IsGlobalPooling()
42 bool IsGlobalAveragePooling(const Pooling2DAttributes& attr, in IsGlobalAveragePooling()
59 absl::any_cast<const Pooling2DAttributes&>(node->operation.attributes); in ApplyToNode()
|
| D | global_pooling_to_reduce_op_test.cc | 40 Pooling2DAttributes attr; in TEST()
|
| D | merge_padding_with.cc | 96 return absl::make_unique<MergePaddingWith2DOperation<Pooling2DAttributes>>( in NewMergePaddingWithPooling()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | pooling_test_util.cc | 33 Pooling2DAttributes attr; in AveragePoolingTest()
64 Pooling2DAttributes attr; in AveragePoolingNonEmptyPaddingTest()
96 Pooling2DAttributes attr; in MaxPoolingTest()
127 Pooling2DAttributes attr; in MaxPoolingIndicesTest()
|
| D | pooling.h | 29 const Pooling2DAttributes& attr);
|
| D | pooling.cc | 289 const Pooling2DAttributes& attr) { in CreatePooling()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | pooling_test.cc | 53 Pooling2DAttributes attr; in TEST()
83 Pooling2DAttributes attr; in TEST()
109 Pooling2DAttributes attr; in TEST()
|
| D | pooling.cc | 34 absl::Status GenerateMaxPoolingCode(const Pooling2DAttributes& attr, in GenerateMaxPoolingCode()
99 const Pooling2DAttributes& attr, const NodeShader::GenerationContext& ctx, in GenerateAveragePoolingCode()
178 const auto& attr = absl::any_cast<const Pooling2DAttributes&>(ctx.op_attr); in GenerateCode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | operations.h | 155 struct Pooling2DAttributes { struct
215 BHWC CalculateOutputShape(const BHWC& input, const Pooling2DAttributes& attr);
236 const Pooling2DAttributes& attr);
|
| D | operation_parser.h | 63 Pooling2DAttributes* attr);
|
| D | operation_parser.cc | 79 Pooling2DAttributes* attr) { in ParsePoolingAttributes()
|
| D | operations.cc | 296 const Pooling2DAttributes& attr) { in CalculateOutputWithoutStrides()
388 const Pooling2DAttributes& attr) { in CalculateSamePadding()
490 BHWC CalculateOutputShape(const BHWC& input, const Pooling2DAttributes& attr) { in CalculateOutputShape()
761 const Pooling2DAttributes& attr) { in CalculateSamePadding()
|
| D | model_builder.cc | 1357 Pooling2DAttributes attr; in Parse()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | simple_selectors.h | 40 std::unique_ptr<GPUOperation> SelectPooling(const Pooling2DAttributes& attr,
|
| D | simple_selectors.cc | 65 std::unique_ptr<GPUOperation> SelectPooling(const Pooling2DAttributes& attr, in SelectPooling()
|
| D | operation_selector.cc | 449 absl::any_cast<Pooling2DAttributes>(node.operation.attributes); in GPUOperationFromNode()
|