Home
last modified time | relevance | path

Searched refs:DEPTHWISE_CONVOLUTION (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/workgroups/
Dideal_workgroup_picker.cc74 {OperationType::DEPTHWISE_CONVOLUTION, uint3(4, 4, 8)},
81 {OperationType::DEPTHWISE_CONVOLUTION, HW(1, 1), HW(1, 1),
83 {OperationType::DEPTHWISE_CONVOLUTION, HW(3, 3), HW(2, 2),
91 {OperationType::DEPTHWISE_CONVOLUTION, uint3(8, 8, 2)},
98 {OperationType::DEPTHWISE_CONVOLUTION, HW(1, 1), HW(1, 1),
100 {OperationType::DEPTHWISE_CONVOLUTION, HW(3, 3), HW(2, 2),
108 {OperationType::DEPTHWISE_CONVOLUTION, uint3(8, 4, 4)},
115 {OperationType::DEPTHWISE_CONVOLUTION, HW(1, 1), HW(1, 1),
117 {OperationType::DEPTHWISE_CONVOLUTION, HW(3, 3), HW(2, 2),
125 {OperationType::DEPTHWISE_CONVOLUTION, uint3(8, 8, 2)},
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Ddepthwise_conv_test.cc64 {ToString(OperationType::DEPTHWISE_CONVOLUTION), std::move(attr)}, in TEST()
102 {ToString(OperationType::DEPTHWISE_CONVOLUTION), std::move(attr)}, in TEST()
141 {ToString(OperationType::DEPTHWISE_CONVOLUTION), std::move(attr)}, in TEST()
Ddepthwise_conv.cc144 *ctx.gpu_info, OperationType::DEPTHWISE_CONVOLUTION, in GenerateCode()
Dregistry.cc81 insert_op(Type::DEPTHWISE_CONVOLUTION, NewDepthwiseConvolutionNodeShader); in Registry()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dadd_bias.cc72 ToString(OperationType::DEPTHWISE_CONVOLUTION)) { in ApplyToNode()
Dfuse_add_to_conv.cc86 ToString(OperationType::DEPTHWISE_CONVOLUTION)) { in ApplyToNodesSequence()
Dfuse_mul_to_conv.cc77 ToString(OperationType::DEPTHWISE_CONVOLUTION)) { in ApplyToNodesSequence()
141 ToString(OperationType::DEPTHWISE_CONVOLUTION)) { in ApplyToNodesSequence()
Dmerge_padding_with.cc110 OperationType::DEPTHWISE_CONVOLUTION); in NewMergePaddingWithDepthwiseConvolution()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dspecial_selector.cc41 OperationType::DEPTHWISE_CONVOLUTION) { in TryDepthwiseConvPlus1x1Conv()
Doperation_selector.cc397 case OperationType::DEPTHWISE_CONVOLUTION: { in GPUOperationFromNode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h47 DEPTHWISE_CONVOLUTION, enumerator
Doperations.cc99 case OperationType::DEPTHWISE_CONVOLUTION: in ToString()
211 {"depthwise_convolution", OperationType::DEPTHWISE_CONVOLUTION}, in OperationTypeFromString()
Dmodel_builder.cc474 node->operation.type = ToString(OperationType::DEPTHWISE_CONVOLUTION); in Parse()
/external/tensorflow/tensorflow/lite/schema/
Dschema_v0.fbs58 DEPTHWISE_CONVOLUTION = 2,
Dschema_v1.fbs63 DEPTHWISE_CONVOLUTION = 2,