Home
last modified time | relevance | path

Searched defs:src_depth (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_constants.h38 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsForConvConstants() local
75 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsForConvConstantsDot() local
113 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeightsForConvConstants() local
H A Dconv_constants.cc119 const int src_depth = DivideRoundUp(weights_shape.i, 4); in GenerateConvolutionConstantCode() local
215 const int src_depth = DivideRoundUp(src_channels, 4); in IsDotConvBetter() local
241 const int src_depth = DivideRoundUp(w_shape.i, 4); in IsConvConstantsSupported() local
H A Dconv_generic.cc1457 const GpuInfo& gpu_info, const OperationDef& definition, int src_depth, in GuessBestParamsApple()
1470 const GpuInfo& gpu_info, const OperationDef& definition, int src_depth, in GuessBestParams()
1826 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in GuessBestParams() local
1843 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in GuessBestParams() local
1879 const int src_depth = DivideRoundUp(weights_shape.c, 4); in GuessBestParams() local
1894 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in GuessBestParams() local
1908 const int src_depth = DivideRoundUp(weights_shape.i, 4); in GuessBestParamsPointwise() local
H A Dconvolution_transposed_thin.h73 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadData() local
112 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsData() local
H A Ddepthwise_conv_3x3.h83 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeightsAndBiases() local
122 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsAndBiasesData() local
H A Ddepthwise_conv_3x3_stride_h2.h76 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeightsAndBiases() local
115 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsAndBiasesData() local
H A Dfully_connected.h95 const int src_depth = DivideRoundUp(src_channels, 4); in RearrangeFCWeightsToOIO4I4() local
159 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeights() local
H A Dfully_connected.cc40 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeFCWeightsToOIO4I4() local
200 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadQuantizedWeights() local
H A Dconcat_z.cc75 const int src_depth = DivideRoundUp(channels[i], 4); in GetConcatKernelCode() local
H A Dconvolution_transposed_thin.cc53 const OperationDef& op_def, int src_depth, int dst_channels, in GenerateConvolutionTransposedCode()
H A Dconvolution_transposed_3x3_thin.cc78 const OperationDef& op_def, const GpuInfo& gpu_info, int src_depth, in GenerateConvolutionTransposedCode()
H A Dconv_weights_converter_test_util.cc78 const int src_depth = DivideRoundUp(weights.shape.i, 4); in ConvolutionWeightsConverterTest() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Dfc_fc_add.h74 const int src_depth = DivideRoundUp(src_channels, 4); in RearrangeFCWeightsToOIO4I4() local
141 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeights() local
H A Dfc_fc_add.cc38 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeFCWeightsToOIO4I4() local
231 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadQuantizedWeights() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DConv3D.cpp74 const unsigned int src_depth = src.shape()[depth_dim]; in calculate_conv3d() local
129 const unsigned int src_depth = src.shape()[depth_dim]; in calculate_conv3d() local
H A DReductionOperation.cpp192 const unsigned int src_depth = src.shape().z(); in compute_reduction_operation() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
H A Dfully_connected.cc45 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in GenerateCode() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dweights_conversion.cc55 const int src_depth = DivideRoundUp(shape.i, 4); in Get2dResourceSize() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClDirectConv3dKernel.cpp104 const size_t src_depth = src0->dimension(3); in configure() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/elk/
H A Delk_fs_visitor.cpp447 elk_fs_reg src_depth; in emit_single_fb_write() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dconvert.cc136 const int src_depth = DivideRoundUp(shape.i, 4); in ConvertToPHWO4I4() local
/aosp_15_r20/external/mesa3d/src/amd/vulkan/meta/
H A Dradv_meta_blit.c265 uint32_t src_depth = u_minify(src_image->vk.extent.depth, src_iview->vk.base_mip_level); in meta_emit_blit() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Doperation_selector.cc55 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in IsRecommendedForWinograd4x4To6x6() local
/aosp_15_r20/external/deqp/external/openglcts/modules/gl/
H A Dgl4cStencilTexturingTests.cpp1179 GLfloat src_depth = 0.0f; in verifyTexture() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/
H A Dbrw_compile_fs.cpp32 brw_reg src_depth, src_stencil; in brw_emit_single_fb_write() local

12