Home
last modified time | relevance | path

Searched refs:src_depth (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_buffer_1x1.cc82 const BHWC& shape, int src_depth, in GetBestParams() argument
135 int src_depth, int dst_depth) { in GetBestParams() argument
349 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in CreateConvBuffer1x1() local
353 GetBestParams(gpu_info, definition, *shape, src_depth, dst_depth); in CreateConvBuffer1x1()
355 conv_params = GetBestParams(gpu_info, definition, src_depth, dst_depth); in CreateConvBuffer1x1()
367 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in CreateConvBuffer1x1() local
371 GetBestParams(gpu_info, definition, *shape, src_depth, dst_depth); in CreateConvBuffer1x1()
373 conv_params = GetBestParams(gpu_info, definition, src_depth, dst_depth); in CreateConvBuffer1x1()
386 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in CreateConvBuffer1x1Wino4x4To6x6() local
390 GetBestParams(gpu_info, definition, *shape, src_depth, dst_depth); in CreateConvBuffer1x1Wino4x4To6x6()
[all …]
Dconv_constants.h37 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsForConvConstants() local
42 for (int s = 0; s < src_depth; ++s) { in RearrangeWeightsForConvConstants()
74 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsForConvConstantsDot() local
79 for (int s = 0; s < src_depth; ++s) { in RearrangeWeightsForConvConstantsDot()
111 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeightsForConvConstants() local
118 const int aligned_ch_count = use_dot_conv ? weights.shape.o * src_depth * 4 in UploadWeightsForConvConstants()
Dconvolution_transposed_thin.h62 int src_depth, int dst_channels,
70 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadData() local
72 weights.shape.w * weights.shape.h * src_depth * weights.shape.o; in UploadData()
109 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsData() local
114 for (int s = 0; s < src_depth; ++s) { in RearrangeWeightsData()
Dfully_connected.h94 const int src_depth = DivideRoundUp(src_channels, 4); in RearrangeFCWeightsToOIO4I4() local
100 for (int s = 0; s < src_depth; ++s) { in RearrangeFCWeightsToOIO4I4()
150 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeights() local
153 const int elements_count = src_depth * dst_depth * 4; in UploadWeights()
178 desc.size = int2(src_depth * 4, dst_depth); in UploadWeights()
Dconv_powervr.cc1059 const GpuInfo& gpu_info, const OperationDef& definition, int src_depth, in GuessBestParams() argument
1110 if (src_depth % 2 == 0) { in GuessBestParams()
1113 if (src_depth % 4 == 0 && conv_params.block_size.w <= 2) { in GuessBestParams()
1145 if (src_depth % 2 == 0) { in GuessBestParams()
1148 if (src_depth % 4 == 0 && conv_params.block_size.w <= 2) { in GuessBestParams()
1152 if (src_depth % 2 == 0) { in GuessBestParams()
1155 if (src_depth % 4 == 0) { in GuessBestParams()
1158 if (src_depth <= 8) { in GuessBestParams()
1159 conv_params.src_depth_loop_size = src_depth; in GuessBestParams()
1190 if (src_depth % 2 == 0 && src_depth >= 16) { in GuessBestParams()
[all …]
Ddepthwise_conv_3x3_stride_h2.h74 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeightsAndBiases() local
76 int texture_height = src_depth; in UploadWeightsAndBiases()
114 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsAndBiasesData() local
117 for (int s = 0; s < src_depth; ++s) { in RearrangeWeightsAndBiasesData()
Ddepthwise_conv_3x3.h80 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeightsAndBiases() local
82 int texture_height = src_depth; in UploadWeightsAndBiases()
120 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsAndBiasesData() local
123 for (int s = 0; s < src_depth; ++s) { in RearrangeWeightsAndBiasesData()
Dconv_constants.cc117 const int src_depth = DivideRoundUp(weights_shape.i, 4); in GenerateConvolutionConstantCode() local
164 for (int s = 0; s < src_depth; ++s) { in GenerateConvolutionConstantCode()
233 const int src_depth = DivideRoundUp(src_channels, 4); in IsDotConvBetter() local
235 return dst_channels * src_depth < src_channels * dst_depth; in IsDotConvBetter()
252 const int src_depth = DivideRoundUp(w_shape.i, 4); in IsConvConstantsSupported() local
255 use_dot_conv ? w_shape.o * src_depth * 4 : w_shape.i * dst_depth * 4; in IsConvConstantsSupported()
Dconvolution_transposed.h121 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeights() local
125 int texture_height = src_depth * kernel_x * kernel_y; in UploadWeights()
145 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeights() local
151 kernel_x * kernel_y * kernel_z * src_depth * dst_depth * 4; in UploadWeights()
187 int texture_height = src_depth * kernel_x * kernel_y * kernel_z; in UploadWeights()
Dconvolution_transposed_thin.cc53 const OperationDef& op_def, int src_depth, int dst_channels, in GenerateConvolutionTransposedCode() argument
108 for (int i = 1; i < src_depth; ++i) { in GenerateConvolutionTransposedCode()
Dconvolution_transposed_3x3_thin.h77 int src_depth, int dst_depth);
Dconv_buffer_1x1.h128 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeights() local
135 weights.shape.h * weights.shape.w * src_depth * dst_depth_aligned * 4; in UploadWeights()
Dconvolution_transposed_3x3_thin.cc42 const OperationDef& op_def, int src_depth, int dst_depth) { in GenerateConvolutionTransposedCode() argument
108 for (int s = 0; s < src_depth; ++s) { in GenerateConvolutionTransposedCode()
Dconv_powervr.h188 const OperationDef& definition, int src_depth,
279 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeights() local
283 int texture_height = src_depth * kernel_x * kernel_y; in UploadWeights()
Dconv_weights_converter_test_util.cc65 const int src_depth = DivideRoundUp(weights.shape.i, 4); in ConvolutionWeightsConverterTest() local
69 int texture_height = src_depth * kernel_x * kernel_y; in ConvolutionWeightsConverterTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h70 const int src_depth = DivideRoundUp(src_channels, 4); in RearrangeFCWeightsToOIO4I4() local
76 for (int s = 0; s < src_depth; ++s) { in RearrangeFCWeightsToOIO4I4()
127 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeights() local
130 const int elements_count = src_depth * dst_depth * 4; in UploadWeights()
158 desc.size = int2(src_depth * 4, dst_depth); in UploadWeights()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dfully_connected.cc42 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in GenerateCode() local
52 {"src_depth", src_depth}, in GenerateCode()
/external/mesa3d/src/intel/compiler/
Dbrw_fs_visitor.cpp443 fs_reg src_depth, src_stencil; in emit_single_fb_write() local
447 src_depth = frag_depth; in emit_single_fb_write()
449 src_depth = fetch_payload_reg(bld, payload.source_depth_reg); in emit_single_fb_write()
456 color0, color1, src0_alpha, src_depth, dst_depth, src_stencil, in emit_single_fb_write()
Dbrw_fs.cpp4368 const fs_reg &src_depth = inst->src[FB_WRITE_LOGICAL_SRC_SRC_DEPTH]; in lower_fb_write_logical_send() local
4529 if (src_depth.file != BAD_FILE) { in lower_fb_write_logical_send()
4530 sources[length] = src_depth; in lower_fb_write_logical_send()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc49 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in IsRecommendedForWinograd4x4To6x6() local
74 src_depth >= min_depth && dst_depth >= min_depth; in IsRecommendedForWinograd4x4To6x6()
355 const int src_depth = DivideRoundUp(weights_shape.i, 4); in GPUOperationFromNode() local
365 int texture_height = src_depth * kernel_x * kernel_y; in GPUOperationFromNode()
/external/deqp/external/openglcts/modules/gl/
Dgl4cStencilTexturingTests.cpp1184 GLfloat src_depth = 0.0f; in verifyTexture() local
1187 memcpy(&src_depth, src_pixel_data, 4); in verifyTexture()
1189 if (de::abs(src_depth - dst_depth) > 0.0001f) in verifyTexture()
1192 << "], got: " << dst_depth << " expected: " << src_depth in verifyTexture()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dconvert.cc136 const int src_depth = DivideRoundUp(shape.i, 4); in ConvertToPHWO4I4() local
142 for (int ch = 0; ch < src_depth; ++ch) { in ConvertToPHWO4I4()
/external/mesa3d/src/gallium/auxiliary/util/
Du_blitter.c1824 int src_depth = abs(srcbox->depth); in do_blits() local
1825 float src_z_step = src_depth / (float)dstbox->depth; in do_blits()
1844 depth_center_offset = 0.5 / dstbox->depth * src_depth; in do_blits()
1959 int src_depth = src->u.tex.last_layer + 1; in util_blitter_blit_generic() local
1980 box.z >= 0 && box.z < src_depth && in util_blitter_blit_generic()
1983 box.z + box.depth > 0 && box.z + box.depth <= src_depth; in util_blitter_blit_generic()
/external/mesa3d/src/amd/vulkan/
Dradv_meta_blit.c315 uint32_t src_depth = radv_minify(src_iview->image->info.depth, src_iview->base_mip); in meta_emit_blit() local
326 src_offset_0[2] / (float)src_depth, in meta_emit_blit()