Home
last modified time | relevance | path

Searched defs:dst_depth (Results 1 – 25 of 30) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_constants.h37 const int dst_depth = DivideRoundUp(weights.shape.o, 4); in RearrangeWeightsForConvConstants() local
74 const int dst_depth = DivideRoundUp(weights.shape.o, 4); in RearrangeWeightsForConvConstantsDot() local
114 const int dst_depth = DivideRoundUp(weights.shape.o, 4); in UploadWeightsForConvConstants() local
H A Dconv_generic.cc1458 int dst_depth, bool x_kernel_is_1, bool y_kernel_is_1, in GuessBestParamsApple()
1471 int dst_depth, bool x_kernel_is_1, bool y_kernel_is_1, in GuessBestParams()
1825 const int dst_depth = DivideRoundUp(attr.weights.shape.o, 4); in GuessBestParams() local
1842 const int dst_depth = DivideRoundUp(attr.weights.shape.o, 4); in GuessBestParams() local
1878 const int dst_depth = DivideRoundUp(weights_shape.b, 4); in GuessBestParams() local
1893 const int dst_depth = DivideRoundUp(attr.weights.shape.o, 4); in GuessBestParams() local
1907 const int dst_depth = DivideRoundUp(weights_shape.o, 4); in GuessBestParamsPointwise() local
H A Dfully_connected.h97 const int dst_depth = DivideRoundUp(dst_channels, 4); in RearrangeFCWeightsToOIO4I4() local
160 const int dst_depth = DivideRoundUp(weights.shape.o, 4); in UploadWeights() local
H A Dfully_connected.cc41 const int dst_depth = DivideRoundUp(weights.shape.o, 4); in RearrangeFCWeightsToOIO4I4() local
201 const int dst_depth = DivideRoundUp(weights.shape.o, 4); in UploadQuantizedWeights() local
H A Dconv_constants.cc216 const int dst_depth = DivideRoundUp(dst_channels, 4); in IsDotConvBetter() local
242 const int dst_depth = DivideRoundUp(w_shape.o, 4); in IsConvConstantsSupported() local
H A Dconvolution_transposed.cc65 const int dst_depth = DivideRoundUp(attr.weights.shape.o, 4); in ConvolutionTransposed() local
110 const int dst_depth = DivideRoundUp(attr.weights.shape.o, 4); in ConvolutionTransposed() local
H A Dconvolution_transposed_3x3_thin.cc79 int dst_depth) { in GenerateConvolutionTransposedCode()
H A Ddepthwise_conv.h120 const int dst_depth = DivideRoundUp(dst_channels, 4); in RearrangeWeightsForDWConv2D() local
H A Dconv_weights_converter_test_util.cc76 const int dst_depth = AlignByN(DivideRoundUp(weights.shape.o, 4), in ConvolutionWeightsConverterTest() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Dfc_fc_add.h76 const int dst_depth = DivideRoundUp(dst_channels, 4); in RearrangeFCWeightsToOIO4I4() local
142 const int dst_depth = DivideRoundUp(weights.shape.o, 4); in UploadWeights() local
H A Dfc_fc_add.cc39 const int dst_depth = DivideRoundUp(weights.shape.o, 4); in RearrangeFCWeightsToOIO4I4() local
232 const int dst_depth = DivideRoundUp(weights.shape.o, 4); in UploadQuantizedWeights() local
H A Dconv_pointwise.cc183 const int dst_depth = DivideRoundUp(dst_channels, 4); in CreateConvPointwise() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dutil.h41 const int dst_depth = dst.size(); in CopyLinearFLT4() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
H A Dfully_connected.cc46 const int dst_depth = DivideRoundUp(attr.weights.shape.o, 4); in GenerateCode() local
H A Dconv.cc244 auto dst_depth = DivideRoundUp(ctx.output_shapes[0][3], 4); in GenerateCode() local
/aosp_15_r20/external/deqp/external/openglcts/modules/gl/
H A Dgl4cStencilTexturingTests.cpp1158 GLfloat dst_depth = 0.0f; in verifyTexture() local
1178 GLfloat dst_depth = 0.0f; in verifyTexture() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dweights_conversion.cc53 const int dst_depth = in Get2dResourceSize() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClDirectConv3dKernel.cpp108 const size_t dst_depth = dst->dimension(3); in configure() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DConv3D.cpp199 const unsigned int dst_depth = dst.shape()[depth_dim]; in conv3d() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/elk/
H A Delk_fs_visitor.cpp446 const elk_fs_reg dst_depth = fetch_payload_reg(bld, fs_payload().dest_depth_reg); in emit_single_fb_write() local
H A Delk_lower_logical_sends.cpp157 const elk_fs_reg dst_depth = inst->src[FB_WRITE_LOGICAL_SRC_DST_DEPTH]; in lower_fb_write_logical_send() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dconvert.cc135 const int dst_depth = DivideRoundUp(shape.o, 4); in ConvertToPHWO4I4() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Doperation_selector.cc56 const int dst_depth = DivideRoundUp(attr.weights.shape.o, 4); in IsRecommendedForWinograd4x4To6x6() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/
H A Dbrw_compile_fs.cpp31 const brw_reg dst_depth = fetch_payload_reg(bld, s.fs_payload().dest_depth_reg); in brw_emit_single_fb_write() local
/aosp_15_r20/external/mesa3d/src/gallium/drivers/radeonsi/
H A Dsi_test_image_copy_region.c767 int src_width, src_height, src_depth, dst_width, dst_height, dst_depth; in si_test_blit() local

12