| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | depthwise_conv_plus_1x1_conv.cc | 28 const Convolution2DAttributes& conv_attr, in UploadWeights() argument
33 int conv_src_ch_aligned = AlignByN(conv_attr.weights.shape.i, 4); in UploadWeights()
34 int conv_dst_ch_aligned = AlignByN(conv_attr.weights.shape.o, 4); in UploadWeights()
66 if (i < conv_attr.bias.shape.v) { in UploadWeights()
67 gpu_data.push_back(conv_attr.bias.data[i]); in UploadWeights()
79 if (s_ch < conv_attr.weights.shape.i && in UploadWeights()
80 d_ch < conv_attr.weights.shape.o) { in UploadWeights()
82 conv_attr.weights.shape.LinearIndex({d_ch, 0, 0, s_ch}); in UploadWeights()
83 gpu_data.push_back(conv_attr.weights.data[f_index]); in UploadWeights()
236 const Convolution2DAttributes& conv_attr) { in IsDepthwiseConvPlus1x1ConvSupported() argument
[all …]
|
| D | depthwise_conv_plus_1x1_conv.h | 37 const Convolution2DAttributes& conv_attr);
42 const Convolution2DAttributes& conv_attr);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | fuse_mul_to_conv_test.cc | 45 Convolution2DAttributes conv_attr; in TEST() local
46 conv_attr.padding.prepended = HW(0, 0); in TEST()
47 conv_attr.padding.appended = HW(0, 0); in TEST()
48 conv_attr.strides = HW(1, 1); in TEST()
49 conv_attr.dilations = HW(1, 1); in TEST()
50 conv_attr.weights.shape = OHWI(16, 3, 2, 8); in TEST()
51 conv_attr.weights.data.resize(conv_attr.weights.shape.DimensionsProduct()); in TEST()
52 conv_attr.bias.shape = Linear(16); in TEST()
53 conv_attr.bias.data.resize(16); in TEST()
63 conv_node->operation.attributes = conv_attr; in TEST()
[all …]
|
| D | fuse_add_to_conv_test.cc | 45 Convolution2DAttributes conv_attr; in TEST() local
46 conv_attr.padding.prepended = HW(0, 0); in TEST()
47 conv_attr.padding.appended = HW(0, 0); in TEST()
48 conv_attr.strides = HW(1, 1); in TEST()
49 conv_attr.dilations = HW(1, 1); in TEST()
50 conv_attr.weights.shape = OHWI(16, 3, 2, 8); in TEST()
51 conv_attr.weights.data.resize(conv_attr.weights.shape.DimensionsProduct()); in TEST()
52 conv_attr.bias.shape = Linear(16); in TEST()
53 conv_attr.bias.data.resize(16); in TEST()
63 conv_node->operation.attributes = conv_attr; in TEST()
|
| D | merge_padding_with_test.cc | 53 Convolution2DAttributes conv_attr; in TEST() local
54 conv_attr.padding.appended = HW(0, 0); in TEST()
55 conv_attr.padding.prepended = HW(0, 0); in TEST()
56 conv_node->operation.attributes = conv_attr; in TEST()
67 conv_attr = in TEST()
69 EXPECT_EQ(HW(1, 1), conv_attr.padding.prepended); in TEST()
70 EXPECT_EQ(HW(2, 2), conv_attr.padding.appended); in TEST()
98 Convolution2DAttributes conv_attr; in TEST() local
99 conv_attr.padding.appended = HW(0, 0); in TEST()
100 conv_attr.padding.prepended = HW(0, 0); in TEST()
[all …]
|
| D | fuse_add_to_conv.cc | 75 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
78 FuseConvolution2DWithAdd(add_attr, conv_attr); in ApplyToNodesSequence()
81 ConvolutionTransposedAttributes* conv_attr = in ApplyToNodesSequence() local
84 FuseConvolutionTransposedWithAdd(add_attr, conv_attr); in ApplyToNodesSequence()
87 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
90 FuseDepthwiseConvolution2DWithAdd(add_attr, conv_attr); in ApplyToNodesSequence()
93 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence() local
96 FuseFullyConnectedWithAdd(add_attr, conv_attr); in ApplyToNodesSequence()
|
| D | fuse_mul_to_conv.cc | 66 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
69 FuseConvolution2DWithMultiply(mul_attr, conv_attr); in ApplyToNodesSequence()
72 ConvolutionTransposedAttributes* conv_attr = in ApplyToNodesSequence() local
75 FuseConvolutionTransposedWithMultiply(mul_attr, conv_attr); in ApplyToNodesSequence()
78 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
81 FuseDepthwiseConvolution2DWithMultiply(mul_attr, conv_attr); in ApplyToNodesSequence()
84 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence() local
87 FuseFullyConnectedWithMultiply(mul_attr, conv_attr); in ApplyToNodesSequence()
130 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
133 FuseMultiplyWithConvolution2D(mul_attr, conv_attr); in ApplyToNodesSequence()
[all …]
|
| D | make_fully_connected.cc | 59 const auto& conv_attr = absl::any_cast<const Convolution2DAttributes&>( in ApplyToNode() local
61 if (!IsConvEquivalentToFullyConnected(conv_attr)) { in ApplyToNode()
66 fc_attr.weights = conv_attr.weights; in ApplyToNode()
67 fc_attr.bias = conv_attr.bias; in ApplyToNode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | fully_connected_selector.cc | 90 Convolution2DAttributes conv_attr; in SelectFullyConnected() local
91 conv_attr.padding.prepended = HW(0, 0); in SelectFullyConnected()
92 conv_attr.padding.appended = HW(0, 0); in SelectFullyConnected()
93 conv_attr.strides = HW(1, 1); in SelectFullyConnected()
94 conv_attr.dilations = HW(1, 1); in SelectFullyConnected()
95 conv_attr.weights = attr.weights; in SelectFullyConnected()
96 conv_attr.bias = attr.bias; in SelectFullyConnected()
98 CreateConvolutionMetal(op_def, dst_shape, conv_attr, gpu_info); in SelectFullyConnected()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | special_selector.cc | 69 auto conv_attr = in TryDepthwiseConvPlus1x1Conv() local
82 if (!IsDepthwiseConvPlus1x1ConvSupported(op_def, dw_attr, conv_attr)) { in TryDepthwiseConvPlus1x1Conv()
87 auto operation = CreateDepthwiseConvPlus1x1Conv(op_def, dw_attr, conv_attr); in TryDepthwiseConvPlus1x1Conv()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 360 ops::Conv2D::Attrs conv_attr; in VerifyFusedBatchNormGradWithConv2D() local
361 conv_attr = conv_attr.DataFormat("NHWC"); in VerifyFusedBatchNormGradWithConv2D()
363 {1, 1, 1, 1}, "SAME", conv_attr); in VerifyFusedBatchNormGradWithConv2D()
|