| /external/tensorflow/tensorflow/lite/kernels/ |
| D | cpu_backend_gemm.h | 118 const MatrixParams<DstScalar>& dst_params, DstScalar* dst_data, in Gemm() argument
122 ValidateParams(lhs_params, rhs_params, dst_params, params); in Gemm()
123 if (!IsValidGemm(lhs_params, rhs_params, dst_params)) { in Gemm()
141 dst_params.order != Order::kColMajor) { in Gemm()
153 dst_params, dst_data, in Gemm()
159 const bool try_custom_gemv = (dst_params.cols == 1); in Gemm()
164 dst_params, dst_data, params, context)) { in Gemm()
171 dst_params, dst_data, params, context); in Gemm()
178 const MatrixParams<int16_t>& dst_params, int16_t* dst_data, in Gemm() argument
182 ValidateParams(lhs_params, rhs_params, dst_params, params); in Gemm()
[all …]
|
| D | cpu_backend_gemm_eigen.cc | 40 const MatrixParams<float>& dst_params, float* dst_data, in Run() argument
58 EigenMatrixMapColMajorMutable eigen_dst(dst_data, dst_params.rows, in Run()
59 dst_params.cols); in Run()
70 BiasAndClamp(params.clamp_min, params.clamp_max, dst_params.rows, in Run()
71 params.bias, dst_params.rows * dst_params.cols, dst_data); in Run()
|
| D | cpu_backend_gemm_gemmlowp.h | 83 const MatrixParams<DstScalar>& dst_params, DstScalar* dst_data,
93 dst_data, dst_params.rows, dst_params.cols);
98 scale_stage.result_offset_after_shift = dst_params.zero_point;
140 const MatrixParams<DstScalar>& dst_params, DstScalar* dst_data,
153 dst_data, dst_params.rows, dst_params.cols);
163 scale_stage.result_offset_after_shift = dst_params.zero_point;
165 ColVectorMap(params.multiplier_fixedpoint_perchannel, dst_params.rows);
167 ColVectorMap(params.multiplier_exponent_perchannel, dst_params.rows);
183 Run(lhs_params, lhs_data, rhs_params, rhs_data, dst_params, dst_data,
|
| D | cpu_backend_gemm_test.cc | 257 const MatrixParams<DstScalar>& dst_params, std::vector<DstScalar>* dst_data, in PerformGemmThenCompareResultsThenAgainWithClamping() argument
262 Gemm(lhs_params, lhs_data.data(), rhs_params, rhs_data.data(), dst_params, in PerformGemmThenCompareResultsThenAgainWithClamping()
275 Gemm(lhs_params, lhs_data.data(), rhs_params, rhs_data.data(), dst_params, in PerformGemmThenCompareResultsThenAgainWithClamping()
284 Gemm(lhs_params, lhs_data.data(), rhs_params, rhs_data.data(), dst_params, in PerformGemmThenCompareResultsThenAgainWithClamping()
315 const MatrixParams<DstScalar>& dst_params, std::vector<DstScalar>* dst_data, in BisectReasonableMultiplierExponent() argument
331 Gemm(lhs_params, lhs_data.data(), rhs_params, rhs_data.data(), dst_params, in BisectReasonableMultiplierExponent()
337 dst_params, dst_data, params_copy, cpu_backend_context); in BisectReasonableMultiplierExponent()
341 dst_params, dst_data, params_copy, cpu_backend_context); in BisectReasonableMultiplierExponent()
350 const MatrixParams<DstScalar>& dst_params, DstScalar* dst_data, in ReferenceGemm() argument
358 cpu_backend_gemm::detail::MakeRuyMatrix(dst_params, dst_data, &ruy_dst); in ReferenceGemm()
[all …]
|
| D | cpu_backend_gemm_x86.h | 41 const MatrixParams<DstScalar>& dst_params, DstScalar* dst_data, in Run()
51 dst_params, dst_data, params, context); in Run()
59 dst_params, dst_data, in Run()
70 const MatrixParams<float>& dst_params, float* dst_data,
75 dst_params, dst_data, params, context);
|
| D | cpu_backend_gemm_custom_gemv.h | 82 const MatrixParams<DstScalar>& dst_params, in IsSupportedGivenSufficientlyManyRows()
91 const MatrixParams<DstScalar>& dst_params, DstScalar* dst_data, in Run()
104 const MatrixParams<DstScalar>& dst_params, DstScalar* dst_data, in CustomGemvTask() argument
111 dst_params_(dst_params), in CustomGemvTask()
150 const MatrixParams<DstScalar>& dst_params, DstScalar* dst_data, in CustomGemv() argument
160 dst_params, params)) { in CustomGemv()
165 context->max_num_threads(), dst_params.rows, dst_params.cols, in CustomGemv()
168 Impl::Run(lhs_params, lhs_data, rhs_params, rhs_data, dst_params, dst_data, in CustomGemv()
176 RoundUp<Impl::kKernelRows>(CeilQuotient(dst_params.rows, thread_count)); in CustomGemv()
179 int row_end = std::min(dst_params.rows, row_start + kRowsPerThread); in CustomGemv()
[all …]
|
| D | cpu_backend_gemm_params.h | 234 const MatrixParams<DstScalar>& dst_params,
245 const MatrixParams<DstScalar>& dst_params) {
251 valid &= dst_params.rows >= 1;
252 valid &= dst_params.cols >= 1;
254 valid &= rhs_params.cols == dst_params.cols;
|
| D | cpu_backend_gemm_eigen.h | 31 const MatrixParams<float>& dst_params, float* dst_data,
|
| D | cpu_backend_gemm_ruy.h | 128 const MatrixParams<DstScalar>& dst_params, DstScalar* dst_data,
136 MakeRuyMatrix(dst_params, dst_data, &ruy_dst);
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | fully_connected.h | 81 cpu_backend_gemm::MatrixParams<DstScalar> dst_params; in FullyConnectedPerChannel() local
82 dst_params.rows = filter_rows; in FullyConnectedPerChannel()
83 dst_params.cols = batches; in FullyConnectedPerChannel()
84 dst_params.order = cpu_backend_gemm::Order::kColMajor; in FullyConnectedPerChannel()
85 dst_params.zero_point = output_offset; in FullyConnectedPerChannel()
96 dst_params, output_data, gemm_params, in FullyConnectedPerChannel()
153 cpu_backend_gemm::MatrixParams<DstScalar> dst_params; in FullyConnected() local
154 dst_params.rows = filter_rows; in FullyConnected()
155 dst_params.cols = batches; in FullyConnected()
156 dst_params.order = cpu_backend_gemm::Order::kColMajor; in FullyConnected()
[all …]
|
| D | conv.h | 108 cpu_backend_gemm::MatrixParams<DstScalar> dst_params; in ConvPerChannel() local
109 dst_params.rows = output_rows; in ConvPerChannel()
110 dst_params.cols = output_cols; in ConvPerChannel()
111 dst_params.order = cpu_backend_gemm::Order::kColMajor; in ConvPerChannel()
112 dst_params.zero_point = output_offset; in ConvPerChannel()
123 dst_params, output_data, gemm_params, in ConvPerChannel()
|
| D | transpose_conv.h | 85 cpu_backend_gemm::MatrixParams<int32_t> dst_params; in TransposeConvV2() local
86 dst_params.order = cpu_backend_gemm::Order::kColMajor; in TransposeConvV2()
87 dst_params.rows = hwoi_ordered_filter_total_size; in TransposeConvV2()
88 dst_params.cols = input_image_size; in TransposeConvV2()
92 input_data + input_offset * i, dst_params, in TransposeConvV2()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | batch_matmul.h | 90 MatrixParams<float> dst_params; in BatchMatMul() local
91 dst_params.order = cpu_backend_gemm::Order::kColMajor; in BatchMatMul()
92 dst_params.rows = lhs_rows; in BatchMatMul()
93 dst_params.cols = rhs_cols; in BatchMatMul()
109 dst_params, out_ptr, gemm_params, context); in BatchMatMul()
200 MatrixParams<int32_t> dst_params; in BatchMatMul() local
201 dst_params.order = cpu_backend_gemm::Order::kColMajor; in BatchMatMul()
202 dst_params.rows = lhs_rows; in BatchMatMul()
203 dst_params.cols = rhs_cols; in BatchMatMul()
228 dst_params, accum_scratch, gemm_params, context); in BatchMatMul()
[all …]
|
| D | sse_tensor_utils.cc | 397 MatrixParams<int32_t> dst_params; in SseCpuBackendGemm() local
398 dst_params.order = cpu_backend_gemm::Order::kColMajor; in SseCpuBackendGemm()
399 dst_params.rows = n_output; in SseCpuBackendGemm()
400 dst_params.cols = n_batch; in SseCpuBackendGemm()
407 dst_params, scratch, gemm_params, context); in SseCpuBackendGemm()
|
| D | optimized_ops.h | 290 cpu_backend_gemm::MatrixParams<float> dst_params; in FullyConnected() local
291 dst_params.order = cpu_backend_gemm::Order::kColMajor; in FullyConnected()
292 dst_params.rows = output_shape.Dims(output_shape.DimensionsCount() - 1); in FullyConnected()
293 dst_params.cols = in FullyConnected()
300 dst_params, output_data, gemm_params, in FullyConnected()
351 cpu_backend_gemm::MatrixParams<uint8> dst_params; in FullyConnected() local
352 dst_params.rows = filter_rows; in FullyConnected()
353 dst_params.cols = batches; in FullyConnected()
354 dst_params.order = cpu_backend_gemm::Order::kColMajor; in FullyConnected()
355 dst_params.zero_point = output_offset; in FullyConnected()
[all …]
|
| D | neon_tensor_utils.cc | 1024 MatrixParams<int32_t> dst_params; in NeonCpuBackendGemm() local
1025 dst_params.order = cpu_backend_gemm::Order::kColMajor; in NeonCpuBackendGemm()
1026 dst_params.rows = n_output; in NeonCpuBackendGemm()
1027 dst_params.cols = n_batch; in NeonCpuBackendGemm()
1034 dst_params, scratch, gemm_params, context); in NeonCpuBackendGemm()
|
| /external/crosvm/devices/src/virtio/video/encoder/ |
| D | mod.rs | 88 dst_params: Params, field
148 let mut dst_params = Params { in new() localVariable
160 .populate_dst_params(&mut dst_params, desired_format, DEFAULT_BUFFER_SIZE) in new()
164 let dest_format = dst_params.format.ok_or(VideoError::InvalidArgument)?; in new()
179 dst_params, in new()
215 dst_params: self.dst_params.clone(), in set_encode_session()
219 frame_rate: self.dst_params.frame_rate, in set_encode_session()
267 self.dst_params.plane_formats[0].plane_size = output_buffer_size; in require_input_buffers()
323 params: self.dst_params.clone(), in require_input_buffers()
712 let resource = match stream.dst_params.resource_type { in resource_create()
[all …]
|
| D | encoder.rs | 52 pub dst_params: Params, field
105 dst_params: &mut Params, in populate_dst_params()
119 dst_params.format = Some(format_desc.format); in populate_dst_params()
123 dst_params.plane_formats = vec { in MatchFunctionParamIds()
1591 dst_params.push_back(param->result_id()); in MatchFunctionParamIds()
1596 src_params, dst_params, "", &Differ::GetSanitizedName, in MatchFunctionParamIds()
1610 src_params, dst_params, 0, &Differ::GroupIdsHelperGetTypeId, in MatchFunctionParamIds()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/diff/ |
| D | diff.cpp | 1586 IdGroup dst_params; in MatchFunctionParamIds() local
1593 [&dst_params](const opt::Instruction* param) { in MatchFunctionParamIds()
1594 dst_params.push_back(param->result_id()); in MatchFunctionParamIds()
1599 src_params, dst_params, "", &Differ::GetSanitizedName, in MatchFunctionParamIds()
1613 src_params, dst_params, 0, &Differ::GroupIdsHelperGetTypeId, in MatchFunctionParamIds()
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/diff/ |
| D | diff.cpp | 1682 IdGroup dst_params; in MatchFunctionParamIds() local
1689 [&dst_params](const opt::Instruction* param) { in MatchFunctionParamIds()
1690 dst_params.push_back(param->result_id()); in MatchFunctionParamIds()
1695 src_params, dst_params, "", &Differ::GetSanitizedName, in MatchFunctionParamIds()
1708 src_params, dst_params, &Differ::GroupIdsHelperGetTypeId, in MatchFunctionParamIds()
|