Home
last modified time | relevance | path

Searched defs:conv_attr (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dfuse_add_to_conv.cc75 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
81 ConvolutionTransposedAttributes* conv_attr = in ApplyToNodesSequence() local
87 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
93 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence() local
Dfuse_mul_to_conv.cc66 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
72 ConvolutionTransposedAttributes* conv_attr = in ApplyToNodesSequence() local
78 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
84 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence() local
130 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
136 ConvolutionTransposedAttributes* conv_attr = in ApplyToNodesSequence() local
142 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
148 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence() local
Dmerge_padding_with_test.cc53 Convolution2DAttributes conv_attr; in TEST() local
98 Convolution2DAttributes conv_attr; in TEST() local
Dmake_fully_connected.cc59 const auto& conv_attr = absl::any_cast<const Convolution2DAttributes&>( in ApplyToNode() local
Dfuse_mul_to_conv_test.cc45 Convolution2DAttributes conv_attr; in TEST() local
102 Convolution2DAttributes conv_attr; in TEST() local
Dfuse_add_to_conv_test.cc45 Convolution2DAttributes conv_attr; in TEST() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Ddepthwise_conv_plus_1x1_conv.cc28 const Convolution2DAttributes& conv_attr, in UploadWeights()
236 const Convolution2DAttributes& conv_attr) { in IsDepthwiseConvPlus1x1ConvSupported()
256 const Convolution2DAttributes& conv_attr) { in CreateDepthwiseConvPlus1x1Conv()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dfully_connected_selector.cc90 Convolution2DAttributes conv_attr; in SelectFullyConnected() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dspecial_selector.cc69 auto conv_attr = in TryDepthwiseConvPlus1x1Conv() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op_test.cc360 ops::Conv2D::Attrs conv_attr; in VerifyFusedBatchNormGradWithConv2D() local