Home
last modified time | relevance | path

Searched refs:CONVOLUTION_2D (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/workgroups/
Dideal_workgroup_picker.cc73 {OperationType::CONVOLUTION_2D, uint3(4, 8, 4)},
79 {OperationType::CONVOLUTION_2D, HW(1, 1), HW(1, 1), uint3(4, 8, 4)},
80 {OperationType::CONVOLUTION_2D, HW(3, 3), HW(2, 2), uint3(8, 4, 4)},
90 {OperationType::CONVOLUTION_2D, uint3(8, 2, 2)},
96 {OperationType::CONVOLUTION_2D, HW(1, 1), HW(1, 1), uint3(4, 2, 8)},
97 {OperationType::CONVOLUTION_2D, HW(3, 3), HW(2, 2), uint3(8, 2, 8)},
107 {OperationType::CONVOLUTION_2D, uint3(8, 4, 4)},
113 {OperationType::CONVOLUTION_2D, HW(1, 1), HW(1, 1), uint3(4, 2, 8)},
114 {OperationType::CONVOLUTION_2D, HW(3, 3), HW(2, 2), uint3(8, 2, 8)},
124 {OperationType::CONVOLUTION_2D, uint3(8, 4, 8)},
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dmake_fully_connected_test.cc65 conv1x1_node0->operation.type = ToString(OperationType::CONVOLUTION_2D); in TEST()
68 conv4x4_node1->operation.type = ToString(OperationType::CONVOLUTION_2D); in TEST()
71 conv1x1_node2->operation.type = ToString(OperationType::CONVOLUTION_2D); in TEST()
99 ASSERT_EQ(ToString(OperationType::CONVOLUTION_2D), in TEST()
101 ASSERT_EQ(ToString(OperationType::CONVOLUTION_2D), in TEST()
Dfuse_mul_to_conv_test.cc62 conv_node->operation.type = ToString(OperationType::CONVOLUTION_2D); in TEST()
87 EXPECT_EQ(ToString(OperationType::CONVOLUTION_2D), in TEST()
113 conv_node->operation.type = ToString(OperationType::CONVOLUTION_2D); in TEST()
138 EXPECT_EQ(ToString(OperationType::CONVOLUTION_2D), in TEST()
Dfuse_add_to_conv_test.cc62 conv_node->operation.type = ToString(OperationType::CONVOLUTION_2D); in TEST()
87 EXPECT_EQ(ToString(OperationType::CONVOLUTION_2D), in TEST()
Dmake_fully_connected.cc46 if (node->operation.type != ToString(OperationType::CONVOLUTION_2D)) { in ApplyToNode()
Dmerge_padding_with_test.cc52 conv_node->operation.type = ToString(OperationType::CONVOLUTION_2D); in TEST()
97 conv_node->operation.type = ToString(OperationType::CONVOLUTION_2D); in TEST()
Dadd_bias.cc55 if (node->operation.type == ToString(OperationType::CONVOLUTION_2D)) { in ApplyToNode()
Dfuse_add_to_conv.cc74 if (conv_node.operation.type == ToString(OperationType::CONVOLUTION_2D)) { in ApplyToNodesSequence()
Dfuse_mul_to_conv.cc65 if (conv_node.operation.type == ToString(OperationType::CONVOLUTION_2D)) { in ApplyToNodesSequence()
129 if (conv_node.operation.type == ToString(OperationType::CONVOLUTION_2D)) { in ApplyToNodesSequence()
Dmerge_padding_with.cc103 OperationType::CONVOLUTION_2D); in NewMergePaddingWithConvolution2D()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconv_test.cc63 {ToString(OperationType::CONVOLUTION_2D), std::move(attr)}, {input}, in TEST()
101 {ToString(OperationType::CONVOLUTION_2D), std::move(attr)}, {input}, in TEST()
138 {ToString(OperationType::CONVOLUTION_2D), std::move(attr)}, {input}, in TEST()
175 {ToString(OperationType::CONVOLUTION_2D), std::move(attr)}, {input}, in TEST()
213 {ToString(OperationType::CONVOLUTION_2D), std::move(attr)}, {input}, in TEST()
Dregistry.cc78 insert_op(Type::CONVOLUTION_2D, NewConvolution1x1NodeShader); in Registry()
79 insert_op(Type::CONVOLUTION_2D, NewConvolutionNodeShader); in Registry()
Dconv.cc137 *ctx.gpu_info, OperationType::CONVOLUTION_2D, in GenerateCode()
278 *ctx.gpu_info, OperationType::CONVOLUTION_2D, in GenerateCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dspecial_selector.cc61 OperationType::CONVOLUTION_2D) { in TryDepthwiseConvPlus1x1Conv()
Doperation_selector.cc271 case OperationType::CONVOLUTION_2D: { in GPUOperationFromNode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h43 CONVOLUTION_2D, enumerator
Doperations.cc91 case OperationType::CONVOLUTION_2D: in ToString()
207 {"convolution_2d", OperationType::CONVOLUTION_2D}, in OperationTypeFromString()
Dmodel_builder.cc378 node->operation.type = ToString(OperationType::CONVOLUTION_2D); in Parse()
790 node->operation.type = ToString(OperationType::CONVOLUTION_2D); in Parse()