| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | transpose_conv.h | 75 for (int out_channel = 0; out_channel < output_depth; in TransposeConv() local
76 ++out_channel) { in TransposeConv()
86 filter_data[Offset(filter_shape, out_channel, filter_y, in TransposeConv()
89 out_channel)] += in TransposeConv()
103 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in TransposeConv() local
105 out_channel)]; in TransposeConv()
107 acc += bias_data[out_channel]; in TransposeConv()
110 acc, output_multiplier[out_channel], output_shift[out_channel]); in TransposeConv()
114 output_data[Offset(output_shape, batch, out_y, out_x, out_channel)] = in TransposeConv()
172 for (int out_channel = 0; out_channel < output_depth; in TransposeConv() local
[all …]
|
| D | conv.h | 69 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in ConvPerChannel() local
89 filter_shape, out_channel, filter_y, filter_x, in_channel)]; in ConvPerChannel()
112 acc += bias_data[out_channel]; in ConvPerChannel()
115 acc, output_multiplier[out_channel], output_shift[out_channel]); in ConvPerChannel()
119 output_data[Offset(output_shape, batch, out_y, out_x, out_channel)] = in ConvPerChannel()
172 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in ConvPerChannel() local
192 filter_shape, out_channel, filter_y, filter_x, in_channel)]; in ConvPerChannel()
204 acc += bias_data[out_channel]; in ConvPerChannel()
207 acc, output_multiplier[out_channel], output_shift[out_channel]); in ConvPerChannel()
210 output_data[Offset(output_shape, batch, out_y, out_x, out_channel)] = in ConvPerChannel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | transpose_conv.h | 76 for (int out_channel = 0; out_channel < output_depth; in TransposeConv() local
77 ++out_channel) { in TransposeConv()
87 filter_data[Offset(filter_shape, out_channel, filter_y, in TransposeConv()
90 out_channel)] += in TransposeConv()
104 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in TransposeConv() local
106 out_channel)] += bias_data[out_channel]; in TransposeConv()
167 for (int out_channel = 0; out_channel < output_depth; in TransposeConv() local
168 ++out_channel) { in TransposeConv()
178 filter_data[Offset(filter_shape, out_channel, filter_y, in TransposeConv()
181 out_channel)] += in TransposeConv()
[all …]
|
| D | conv.h | 62 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in Conv() local
82 filter_shape, out_channel, filter_y, filter_x, in_channel)]; in Conv()
89 bias_value = bias_data[out_channel]; in Conv()
91 output_data[Offset(output_shape, batch, out_y, out_x, out_channel)] = in Conv()
145 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in Conv() local
165 filter_shape, out_channel, filter_y, filter_x, in_channel)]; in Conv()
172 acc += bias_data[out_channel]; in Conv()
179 output_data[Offset(output_shape, batch, out_y, out_x, out_channel)] = in Conv()
223 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in HybridConvPerChannel() local
240 filter_data[Offset(filter_shape, out_channel, filter_y, in HybridConvPerChannel()
[all …]
|
| D | conv3d.h | 61 for (int out_channel = 0; out_channel < output_num_channels; in Conv3D() local
62 ++out_channel) { in Conv3D()
89 filter_x, in_channel, out_channel)]; in Conv3D()
97 bias_value = bias_data[out_channel]; in Conv3D()
100 out_channel)] = in Conv3D()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | concat_z.cc | 85 int out_channel = 0; in GetConcatKernelCode() local
98 c += " result" + postfix[out_channel] + " = "; in GetConcatKernelCode()
100 out_channel++; in GetConcatKernelCode()
101 if (out_channel == 4) { in GetConcatKernelCode()
102 out_channel = 0; in GetConcatKernelCode()
111 if (out_channel != 0) { in GetConcatKernelCode()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | conv.cc | 76 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in ConvPerChannel() local
110 ((out_channel * filter_height + filter_y) * filter_width + in ConvPerChannel()
135 ae_q56s bias_56 = AE_CVTQ48A32S(bias_data[out_channel]); in ConvPerChannel()
148 acc_24x2, output_multiplier[out_channel] >> 8, in ConvPerChannel()
149 output_shift[out_channel]); in ConvPerChannel()
159 out_channel; in ConvPerChannel()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | conv2d_benchmark.py | 189 dtype, data_format, in_channel, out_channel, hw_stride, padding = args
193 batch_size = out_channel
203 fshape = [fh, fw, in_channel, out_channel]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_using_gemm.cc | 134 for (int out_channel = 0; out_channel < filter_count; ++out_channel) { in operator ()() local
180 (in_channel * filter_count) + out_channel]; in operator ()()
187 (out_x * filter_count) + out_channel] = total; in operator ()()
|
| D | depthwise_conv_op_gpu.h | 109 const int out_channel = thread_id % out_depth;
114 const int in_channel = out_channel / depth_multiplier;
115 const int multiplier = out_channel % depth_multiplier;
364 const int out_channel = (thread_id / out_width / out_height) % out_depth;
370 const int in_channel = out_channel / depth_multiplier;
371 const int multiplier = out_channel % depth_multiplier;
919 UNROLL for (int out_channel = out_channel_start;
920 out_channel < out_channel_end; ++out_channel) {
924 const int filter_dm = out_channel - out_channel_start;
936 (out_channel * out_height * out_width) + (out_row * out_width) +
[all …]
|
| D | quantized_conv_ops.cc | 109 for (int out_channel = 0; out_channel < filter_count; ++out_channel) { in operator ()() local
159 (in_channel * filter_count) + out_channel]; in operator ()()
178 (out_x * filter_count) + out_channel] = clamped_output; in operator ()()
|
| /external/llvm-project/llvm/bindings/ocaml/bitwriter/ |
| D | llvm_bitwriter.mli | 36 val output_bitcode : ?unbuffered:bool -> out_channel -> Llvm.llmodule -> bool
|
| /external/llvm/bindings/ocaml/bitwriter/ |
| D | llvm_bitwriter.mli | 37 val output_bitcode : ?unbuffered:bool -> out_channel -> Llvm.llmodule -> bool
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 277 int64_t out_channel = filter_shape[3]; in HandleConv2DFilter() local
295 block_size, channel, out_channel}; in HandleConv2DFilter()
305 channel * block_size * block_size, out_channel}; in HandleConv2DFilter()
|