Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dfully_connected_texture.h76 const int src_depth = AlignByN(IntegralDivideRoundUp(weights.shape.i, 4), 4); in UploadWeights() local
95 const int src_depth = AlignByN(IntegralDivideRoundUp(weights.shape.i, 4), 4); in RearrangeWeightsFP16() local
142 const int src_depth = AlignByN(IntegralDivideRoundUp(weights.shape.i, 4), 4); in RearrangeWeightsFP32() local
Dconvolution_transposed_thin.h85 const int src_depth = IntegralDivideRoundUp(src_channels_, 4); in UploadWeights() local
107 const int src_depth = IntegralDivideRoundUp(src_channels_, 4); in RearrangeWeightsData() local
Ddepth_wise_conv_3x3.h86 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in UploadWeightsAndBiases() local
132 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in RearrangeWeightsAndBiasesData() local
Dconvolution_transposed_3x3_thin.h85 const int src_depth = IntegralDivideRoundUp(src_channels_, 4); in UploadWeights() local
111 const int src_depth = IntegralDivideRoundUp(src_channels_, 4); in RearrangeWeightsData() local
Dconvolution_transposed_4x4.h85 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in UploadWeights() local
110 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in RearrangeWeightsData() local
Dconvolution_transposed.h95 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in UploadWeights() local
164 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in RearrangeWeightsData() local
Dconv_texture.h119 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in UploadWeights() local
175 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in RearrangeWeightsData() local
Dconvolution_transposed_3d.h95 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in UploadWeights() local
166 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in RearrangeWeightsData() local
Dconv_buffer.h85 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in UploadWeights() local
Dconv_buffer_1x1.h109 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in UploadWeights() local
Dconv_powervr.cc533 const CLDevice& device, const OperationDef& definition, int src_depth, in GuessBestParams()
648 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in GuessBestParams() local
665 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in GuessBestParams() local
Dconv_powervr.h142 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in UploadWeights() local
Dconv_constants.h117 const int src_depth = IntegralDivideRoundUp(weights.shape.i, 4); in RearrangeWeightsData() local
Dconvolution_transposed_3x3_thin.cc31 const OperationDef& op_def, const LinearStorage& biases, int src_depth, in GenerateConvolutionTransposedCode()
Dconvolution_transposed_thin.cc32 const OperationDef& op_def, int src_depth, int dst_channels, in GenerateConvolutionTransposedCode()
Dconv_constants.cc45 const int src_depth = IntegralDivideRoundUp(src_channels, 4); in GenerateConvolutionConstantCode() local
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtranspose_conv.cc133 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in GetDeconvolution() local
267 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in GetDeconvolutionShared() local
499 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in GetDeconvolutionBorder() local
662 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in GetDeconvolution3x3() local
882 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in GetDeconvolutionShared3x3() local
904 const int src_depth = IntegralDivideRoundUp(params.weights.shape.i, 4); in ConvolutionTransposed() local
1018 const int src_depth = IntegralDivideRoundUp(params.weights.shape.i, 4); in ConvolutionTransposed3x3() local
Dfully_connected.cc42 const int src_depth = IntegralDivideRoundUp(src_channels, 4); in GetFullyConnectedCode() local
141 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in FullyConnected() local
Dsoftmax.cc192 const int src_depth = IntegralDivideRoundUp(channels_count, 4); in Softmax1x1() local
Ddepthwise_conv.cc211 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in ReorderWeightsDepthWiseConv3x3Stride1x1() local
406 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in ReorderWeightsDepthWiseConv3x3Stride2() local
Dconv.cc160 const int src_depth = IntegralDivideRoundUp(params.weights.shape.i, 4); in GetKernelForConv() local
753 const int src_depth = IntegralDivideRoundUp(params.weights.shape.i, 4); in ReorderWeightsForConv() local
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dfully_connected.cc42 const int src_depth = IntegralDivideRoundUp(attr.weights.shape.i, 4); in GenerateCode() local
/external/mesa3d/src/intel/compiler/
Dbrw_fs_visitor.cpp443 fs_reg src_depth, src_stencil; in emit_single_fb_write() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dconvert.cc127 const int src_depth = IntegralDivideRoundUp(shape.i, 4); in ConvertToPHWO4I4() local
/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

12