Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_constants.h38 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsForConvConstants() local
43 for (int s = 0; s < src_depth; ++s) { in RearrangeWeightsForConvConstants()
75 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsForConvConstantsDot() local
80 for (int s = 0; s < src_depth; ++s) { in RearrangeWeightsForConvConstantsDot()
113 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeightsForConvConstants() local
120 const int aligned_ch_count = use_dot_conv ? weights.shape.o * src_depth * 4 in UploadWeightsForConvConstants()
Dconvolution_transposed_thin.h65 int src_depth, int dst_channels,
73 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadData() local
75 weights.shape.w * weights.shape.h * src_depth * weights.shape.o; in UploadData()
112 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsData() local
117 for (int s = 0; s < src_depth; ++s) { in RearrangeWeightsData()
Dfully_connected.h95 const int src_depth = DivideRoundUp(src_channels, 4); in RearrangeFCWeightsToOIO4I4() local
101 for (int s = 0; s < src_depth; ++s) { in RearrangeFCWeightsToOIO4I4()
159 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeights() local
162 const int elements_count = src_depth * dst_depth * 4; in UploadWeights()
196 TensorStorageType::TEXTURE_2D, src_depth * 4, dst_depth, data.data()); in UploadWeights()
Dfully_connected.cc40 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeFCWeightsToOIO4I4() local
45 for (int s = 0; s < src_depth; ++s) { in RearrangeFCWeightsToOIO4I4()
200 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadQuantizedWeights() local
203 std::vector<uint8_t> data(src_depth * 4 * dst_depth * 4); in UploadQuantizedWeights()
206 DataType::UINT8, TensorStorageType::TEXTURE_2D, src_depth * 4, dst_depth, in UploadQuantizedWeights()
Dconv_constants.cc119 const int src_depth = DivideRoundUp(weights_shape.i, 4); in GenerateConvolutionConstantCode() local
150 for (int s = 0; s < src_depth; ++s) { in GenerateConvolutionConstantCode()
215 const int src_depth = DivideRoundUp(src_channels, 4); in IsDotConvBetter() local
217 return dst_channels * src_depth < src_channels * dst_depth; in IsDotConvBetter()
241 const int src_depth = DivideRoundUp(w_shape.i, 4); in IsConvConstantsSupported() local
244 use_dot_conv ? w_shape.o * src_depth * 4 : w_shape.i * dst_depth * 4; in IsConvConstantsSupported()
Dconv_generic.cc1457 const GpuInfo& gpu_info, const OperationDef& definition, int src_depth, in GuessBestParamsApple() argument
1462 y_kernel_is_1, src_depth, dst_shape); in GuessBestParamsApple()
1465 y_kernel_is_1, src_depth, dst_shape); in GuessBestParamsApple()
1470 const GpuInfo& gpu_info, const OperationDef& definition, int src_depth, in GuessBestParams() argument
1521 if (src_depth % 2 == 0) { in GuessBestParams()
1524 if (src_depth % 4 == 0 && conv_params.block_size.w <= 2) { in GuessBestParams()
1553 if (src_depth % 2 == 0) { in GuessBestParams()
1556 if (src_depth % 4 == 0 && conv_params.block_size.w <= 2) { in GuessBestParams()
1560 if (src_depth % 2 == 0) { in GuessBestParams()
1563 if (src_depth % 4 == 0) { in GuessBestParams()
[all …]
Ddepthwise_conv_3x3.h83 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeightsAndBiases() local
85 int texture_height = src_depth; in UploadWeightsAndBiases()
122 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsAndBiasesData() local
125 for (int s = 0; s < src_depth; ++s) { in RearrangeWeightsAndBiasesData()
Ddepthwise_conv_3x3_stride_h2.h76 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeightsAndBiases() local
78 int texture_height = src_depth; in UploadWeightsAndBiases()
115 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeWeightsAndBiasesData() local
118 for (int s = 0; s < src_depth; ++s) { in RearrangeWeightsAndBiasesData()
Dconcat_z.cc75 const int src_depth = DivideRoundUp(channels[i], 4); in GetConcatKernelCode() local
76 if (src_depth % 2 == 0) { in GetConcatKernelCode()
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.h76 int src_depth, int dst_depth);
Dconvolution_transposed_3x3_thin.cc78 const OperationDef& op_def, const GpuInfo& gpu_info, int src_depth, in GenerateConvolutionTransposedCode() argument
115 for (int s = 0; s < src_depth; ++s) { in GenerateConvolutionTransposedCode()
Dconv_generic.h191 const OperationDef& definition, int src_depth,
197 const OperationDef& definition, int src_depth,
Dconv_weights_converter_test_util.cc78 const int src_depth = DivideRoundUp(weights.shape.i, 4); in ConvolutionWeightsConverterTest() local
82 int texture_height = src_depth * kernel_x * kernel_y; in ConvolutionWeightsConverterTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h74 const int src_depth = DivideRoundUp(src_channels, 4); in RearrangeFCWeightsToOIO4I4() local
80 for (int s = 0; s < src_depth; ++s) { in RearrangeFCWeightsToOIO4I4()
141 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadWeights() local
144 const int elements_count = src_depth * dst_depth * 4; in UploadWeights()
177 TensorStorageType::TEXTURE_2D, src_depth * 4, dst_depth, data.data()); in UploadWeights()
Dfc_fc_add.cc38 const int src_depth = DivideRoundUp(weights.shape.i, 4); in RearrangeFCWeightsToOIO4I4() local
43 for (int s = 0; s < src_depth; ++s) { in RearrangeFCWeightsToOIO4I4()
231 const int src_depth = DivideRoundUp(weights.shape.i, 4); in UploadQuantizedWeights() local
234 std::vector<uint8_t> data(src_depth * 4 * dst_depth * 4); in UploadQuantizedWeights()
237 DataType::UINT8, TensorStorageType::TEXTURE_2D, src_depth * 4, dst_depth, in UploadQuantizedWeights()
/external/ComputeLibrary/tests/validation/reference/
DReductionOperation.cpp192 const unsigned int src_depth = src.shape().z(); in compute_reduction_operation() local
236 … const int in_offset = du * src_depth * src_height * src_width + y * src_width + x; in compute_reduction_operation()
252 for(unsigned int z = 0; z < src_depth; ++z) in compute_reduction_operation()
258 …const int in_offset = du * src_batch * src_depth * src_height * src_width + z * src_width * src_… in compute_reduction_operation()
259 …const int out_offset = du * src_depth * src_height * src_width + z * src_width * src_height + y *… in compute_reduction_operation()
262 …e_operation_arg_min_max<T, OT>(src_row_ptr, reduce_elems, op, src_width * src_height * src_depth) : in compute_reduction_operation()
263 …reduce_operation<T, OT>(src_row_ptr, reduce_elems, op, src_width * src_height * src_depth, policy); in compute_reduction_operation()
DConv3D.cpp74 const unsigned int src_depth = src.shape()[depth_dim]; in calculate_conv3d() local
90 const bool is_z_valid = is_valid_pixel(idx_z, 0, src_depth); in calculate_conv3d()
129 const unsigned int src_depth = src.shape()[depth_dim]; in calculate_conv3d() local
160 const bool is_z_valid = is_valid_pixel(idx_z, 0, src_depth); in calculate_conv3d()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dfully_connected.cc45 const int src_depth = DivideRoundUp(attr.weights.shape.i, 4); in GenerateCode() local
55 {"src_depth", src_depth}, in GenerateCode()
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_test_image_copy_region.c767 int src_width, src_height, src_depth, dst_width, dst_height, dst_depth; in si_test_blit() local
788 src_depth = (rand() % max_src_depth) + 1; in si_test_blit()
796 srcz = rand() % (util_num_layers(&tsrc, src_level) - src_depth + 1); in si_test_blit()
835 if (src_depth > dst_depth) in si_test_blit()
836 src_depth = dst_depth; in si_test_blit()
844 if (src_depth < dst_depth) in si_test_blit()
845 dst_depth = src_depth; in si_test_blit()
863 u_box_3d(srcx, srcy, srcz, src_width, src_height, src_depth, &info.src.box); in si_test_blit()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_conversion.cc55 const int src_depth = DivideRoundUp(shape.i, 4); in Get2dResourceSize() local
57 return uint2(dst_depth, src_depth * shape.h * shape.w * shape.d); in Get2dResourceSize()
/external/mesa3d/src/intel/compiler/elk/
Delk_fs_visitor.cpp447 elk_fs_reg src_depth; in emit_single_fb_write() local
450 src_depth = frag_depth; in emit_single_fb_write()
460 src_depth = fetch_payload_reg(bld, fs_payload().source_depth_reg); in emit_single_fb_write()
464 color0, color1, src0_alpha, src_depth, dst_depth, in emit_single_fb_write()
/external/ComputeLibrary/src/gpu/cl/kernels/
DClDirectConv3dKernel.cpp104 const size_t src_depth = src0->dimension(3); in configure() local
130 build_options.add_option("-DSRC_DEPTH=" + support::cpp11::to_string(src_depth)); in configure()
/external/deqp/external/openglcts/modules/gl/
Dgl4cStencilTexturingTests.cpp1179 GLfloat src_depth = 0.0f; in verifyTexture() local
1182 memcpy(&src_depth, src_pixel_data, 4); in verifyTexture()
1184 if (de::abs(src_depth - dst_depth) > 0.0001f) in verifyTexture()
1188 << " expected: " << src_depth << tcu::TestLog::EndMessage; 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()

12