Home
last modified time | relevance | path

Searched refs:in_channels (Results 1 – 25 of 53) sorted by relevance

123

/external/ComputeLibrary/src/core/CL/cl_kernels/nhwc/
Dchannel_shuffle.cl89 …in_channels = (curr_out_channels * (VEC_DATA_TYPE(uint, VEC_SIZE))(K)) % (VEC_DATA_TYPE(uint, VEC_…
95 DATA_TYPE out0 = *((const __global * DATA_TYPE)(input_ptr) + in_channels);
100 *(input_ptr + in_channels.s0),
101 *(input_ptr + in_channels.s1)
107 *(input_ptr + in_channels.s0),
108 *(input_ptr + in_channels.s1),
109 *(input_ptr + in_channels.s2)
115 *(input_ptr + in_channels.s0),
116 *(input_ptr + in_channels.s1),
117 *(input_ptr + in_channels.s2),
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dcudnn_deterministic_base.py103 depth=3, height=3, width=3, in_channels=3, out_channels=2)
123 depth=3, height=3, width=3, in_channels=8, out_channels=1)
138 height=3, width=3, in_channels=8, out_channels=8)
163 height=7, width=7, in_channels=1, out_channels=3)
186 in_channels = 3
189 batch=1, height=16, width=16, channels=in_channels)
191 height=7, width=7, out_channels=out_channels, in_channels=in_channels)
216 in_channels = 8
219 batch=8, height=64, width=64, channels=in_channels)
221 height=3, width=3, out_channels=out_channels, in_channels=in_channels)
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Conv3DBackpropInputV2.pbtxt8 `[batch, depth, rows, cols, in_channels]` tensor.
14 Shape `[depth, rows, cols, in_channels, out_channels]`.
15 `in_channels` must match between `input` and `filter`.
43 [batch, in_depth, in_height, in_width, in_channels].
45 [batch, in_channels, in_depth, in_height, in_width].
Dapi_def_Conv3D.pbtxt6 Shape `[batch, in_depth, in_height, in_width, in_channels]`.
12 Shape `[filter_depth, filter_height, filter_width, in_channels,
13 out_channels]`. `in_channels` must match between `input` and `filter`.
34 [batch, in_depth, in_height, in_width, in_channels].
36 [batch, in_channels, in_depth, in_height, in_width].
Dapi_def_DepthwiseConv2dNative.pbtxt38 Given an input tensor of shape `[batch, in_height, in_width, in_channels]`
40 `[filter_height, filter_width, in_channels, channel_multiplier]`, containing
41 `in_channels` convolutional filters of depth 1, `depthwise_conv2d` applies
44 together. Thus, the output has `in_channels * channel_multiplier` channels.
47 for k in 0..in_channels-1
Dapi_def_Conv2DBackpropFilter.pbtxt6 4-D with shape `[batch, in_height, in_width, in_channels]`.
14 `[filter_height, filter_width, in_channels, out_channels]` tensor.
28 `[filter_height, filter_width, in_channels, out_channels]`. Gradient w.r.t.
60 [batch, in_height, in_width, in_channels].
62 [batch, in_channels, in_height, in_width].
Dapi_def_Conv3DBackpropFilterV2.pbtxt6 Shape `[batch, depth, rows, cols, in_channels]`.
14 `[filter_depth, filter_height, filter_width, in_channels, out_channels]`
43 [batch, in_depth, in_height, in_width, in_channels].
45 [batch, in_channels, in_depth, in_height, in_width].
Dapi_def_Conv3DBackpropInput.pbtxt6 Shape `[batch, depth, rows, cols, in_channels]`.
12 Shape `[depth, rows, cols, in_channels, out_channels]`.
13 `in_channels` must match between `input` and `filter`.
Dapi_def_Conv3DBackpropFilter.pbtxt6 Shape `[batch, depth, rows, cols, in_channels]`.
12 Shape `[depth, rows, cols, in_channels, out_channels]`.
13 `in_channels` must match between `input` and `filter`.
Dapi_def_Conv2D.pbtxt14 `[filter_height, filter_width, in_channels, out_channels]`
69 Given an input tensor of shape `[batch, in_height, in_width, in_channels]`
71 `[filter_height, filter_width, in_channels, out_channels]`, this op
75 `[filter_height * filter_width * in_channels, output_channels]`.
78 filter_height * filter_width * in_channels]`.
Dapi_def_Conv2DBackpropInput.pbtxt14 `[filter_height, filter_width, in_channels, out_channels]`.
27 4-D with shape `[batch, in_height, in_width, in_channels]`. Gradient
59 [batch, in_height, in_width, in_channels].
61 [batch, in_channels, in_height, in_width].
Dapi_def_DepthwiseConv2dNativeBackpropFilter.pbtxt8 in_width, in_channels]` tensor.
16 `[filter_height, filter_width, in_channels, depthwise_multiplier]` tensor.
32 `[filter_height, filter_width, in_channels, out_channels]`. Gradient w.r.t.
Dapi_def_MaxPool.pbtxt39 [batch, in_height, in_width, in_channels].
41 [batch, in_channels, in_height, in_width].
Dapi_def_MaxPoolV2.pbtxt39 [batch, in_height, in_width, in_channels].
41 [batch, in_channels, in_height, in_width].
Dapi_def_BiasAddGrad.pbtxt22 [batch, in_channels, in_height, in_width].
23 The tensor will be added to "in_channels", the third-to-the-last
Dapi_def_BiasAdd.pbtxt28 [batch, in_channels, in_height, in_width].
29 The tensor will be added to "in_channels", the third-to-the-last
Dapi_def_MaxPool3D.pbtxt40 [batch, in_depth, in_height, in_width, in_channels].
42 [batch, in_channels, in_depth, in_height, in_width].
Dapi_def_AvgPool.pbtxt38 [batch, in_height, in_width, in_channels].
40 [batch, in_channels, in_height, in_width].
Dapi_def_AvgPool3DGrad.pbtxt46 [batch, in_depth, in_height, in_width, in_channels].
48 [batch, in_channels, in_depth, in_height, in_width].
Dapi_def_MaxPoolGradGrad.pbtxt51 [batch, in_height, in_width, in_channels].
53 [batch, in_channels, in_height, in_width].
Dapi_def_MaxPoolGrad.pbtxt52 [batch, in_height, in_width, in_channels].
54 [batch, in_channels, in_height, in_width].
/external/tensorflow/tensorflow/core/kernels/
Deigen_spatial_convolutions_test.cc668 const int in_channels = 10; in TEST() local
682 Tensor<float, 4> input(in_channels, in_depth, in_rows, in_cols); in TEST()
683 Tensor<float, 5> kernel(kern_filters, in_channels, kern_depth, kern_height, in TEST()
709 for (int id = 0; id < in_channels; ++id) { in TEST()
729 const int in_channels = 10; in TEST() local
743 Tensor<float, 4, RowMajor> input(in_cols, in_rows, in_depth, in_channels); in TEST()
745 in_channels, kern_filters); in TEST()
771 for (int id = 0; id < in_channels; ++id) { in TEST()
791 const int in_channels = 10; in TEST() local
805 Tensor<float, 4> input(in_channels, in_depth, in_rows, in_cols); in TEST()
[all …]
/external/webrtc/modules/audio_coding/codecs/opus/test/
Dlapped_transform_unittest.cc28 size_t in_channels, in ProcessAudioBlock() argument
32 RTC_CHECK_EQ(in_channels, out_channels); in ProcessAudioBlock()
50 size_t in_channels, in ProcessAudioBlock() argument
54 RTC_CHECK_EQ(in_channels, out_channels); in ProcessAudioBlock()
/external/libaom/av1/encoder/
Dcnn.c203 channels_per_branch[b] = layer_config->in_channels; in find_cnn_out_channels()
371 for (int k = 0; k < layer_config->in_channels; ++k) { in convolve_maxpool_padding_zero()
413 for (int k = 0; k < layer_config->in_channels; ++k) { in convolve_maxpool_padding_replicate()
457 for (int k = 0; k < layer_config->in_channels; ++k) { in convolve_maxpool_padding_valid()
506 for (int k = 0; k < layer_config->in_channels; ++k) { in convolve_element_wise()
552 for (int k = 0; k < layer_config->in_channels; ++k) { in convolve_no_maxpool_padding_zero()
596 for (int k = 0; k < layer_config->in_channels; ++k) { in convolve_no_maxpool_padding_replicate()
636 for (int k = 0; k < layer_config->in_channels; ++k) { in av1_cnn_convolve_no_maxpool_padding_valid_c()
658 const int cstep = layer_config->in_channels * layer_config->out_channels; in av1_cnn_convolve()
792 const int cstep = layer_config->in_channels * layer_config->out_channels; in av1_cnn_deconvolve_c()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dflatten_atrous.cc71 const int32_t in_channels = filter.dim_size(2); in FlattenAtrousConv() local
81 in_channels, out_channels})); in FlattenAtrousConv()
89 for (int c_in = 0; c_in < in_channels; ++c_in) { in FlattenAtrousConv()

123