Home
last modified time | relevance | path

Searched refs:rhs_params (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm.h117 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in Gemm() argument
122 ValidateParams(lhs_params, rhs_params, dst_params, params); in Gemm()
133 rhs_params.order != Order::kColMajor || in Gemm()
145 rhs_params, rhs_data, in Gemm()
156 if (detail::CustomGemv(lhs_params, lhs_data, rhs_params, rhs_data, in Gemm()
163 quantization_flavor>::Run(lhs_params, lhs_data, rhs_params, rhs_data, in Gemm()
172 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in Gemm() argument
177 ValidateParams(lhs_params, rhs_params, dst_params, params); in Gemm()
184 rhs_params, rhs_data, in Gemm()
Dcpu_backend_gemm_gemmlowp.h82 const MatrixParams<SrcScalar>& rhs_params, const SrcScalar* rhs_data,
91 gemmlowp_rhs(rhs_data, rhs_params.rows, rhs_params.cols);
116 &gemmlowp_dst, -lhs_params.zero_point, -rhs_params.zero_point,
123 &gemmlowp_dst, -lhs_params.zero_point, -rhs_params.zero_point,
139 const MatrixParams<SrcScalar>& rhs_params, const SrcScalar* rhs_data,
151 gemmlowp_rhs(rhs_data, rhs_params.rows, rhs_params.cols);
179 -lhs_params.zero_point, -rhs_params.zero_point, output_pipeline);
183 Run(lhs_params, lhs_data, rhs_params, rhs_data, dst_params, dst_data,
Dcpu_backend_gemm_x86.h40 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in Run()
50 quantization_flavor>::Run(lhs_params, lhs_data, rhs_params, rhs_data, in Run()
58 rhs_params, rhs_data, in Run()
69 const MatrixParams<float>& rhs_params, const float* rhs_data,
74 GemmImplUsingEigen::Run(lhs_params, lhs_data, rhs_params, rhs_data,
Dcpu_backend_gemm_test.cc255 const MatrixParams<RhsScalar>& rhs_params, 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()
313 const MatrixParams<RhsScalar>& rhs_params, in BisectReasonableMultiplierExponent() argument
331 Gemm(lhs_params, lhs_data.data(), rhs_params, rhs_data.data(), dst_params, in BisectReasonableMultiplierExponent()
336 bisect_mid + 1, bisect_max, lhs_params, lhs_data, rhs_params, rhs_data, in BisectReasonableMultiplierExponent()
340 bisect_min, bisect_mid, lhs_params, lhs_data, rhs_params, rhs_data, in BisectReasonableMultiplierExponent()
349 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in ReferenceGemm() argument
357 cpu_backend_gemm::detail::MakeRuyMatrix(rhs_params, rhs_data, &ruy_rhs); in ReferenceGemm()
[all …]
Dcpu_backend_gemm_eigen.cc39 const MatrixParams<float>& rhs_params, const float* rhs_data, in Run() argument
56 EigenMatrixMapColMajorConst eigen_rhs(rhs_data, rhs_params.rows, in Run()
57 rhs_params.cols); in Run()
61 if (rhs_params.cols == 1) { in Run()
Dcpu_backend_gemm_custom_gemv.h81 const MatrixParams<RhsScalar>& rhs_params, in IsSupportedGivenSufficientlyManyRows()
90 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in Run()
103 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in CustomGemvTask() argument
109 rhs_params_(rhs_params), in CustomGemvTask()
149 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in CustomGemv() argument
159 if (!Impl::IsSupportedGivenSufficientlyManyRows(lhs_params, rhs_params, in CustomGemv()
168 Impl::Run(lhs_params, lhs_data, rhs_params, rhs_data, dst_params, dst_data, in CustomGemv()
180 tasks.emplace_back(lhs_params, lhs_data, rhs_params, rhs_data, dst_params, in CustomGemv()
311 const MatrixParams<RhsScalar>& rhs_params,
324 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data,
[all …]
Dcpu_backend_gemm_eigen.h30 const MatrixParams<float>& rhs_params, const float* rhs_data,
Dcpu_backend_gemm_ruy.h127 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data,
135 MakeRuyMatrix(rhs_params, rhs_data, &ruy_rhs, context->use_caching());
Dcpu_backend_gemm_params.h233 const MatrixParams<RhsScalar>& rhs_params,
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h69 cpu_backend_gemm::MatrixParams<int8> rhs_params; in FullyConnected() local
70 rhs_params.rows = filter_cols; in FullyConnected()
71 rhs_params.cols = batches; in FullyConnected()
72 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in FullyConnected()
73 rhs_params.zero_point = -input_offset; in FullyConnected()
85 cpu_backend_gemm::Gemm(lhs_params, filter_data, rhs_params, input_data, in FullyConnected()
Dtranspose_conv.h74 cpu_backend_gemm::MatrixParams<int8_t> rhs_params; in TransposeConvV2() local
75 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in TransposeConvV2()
76 rhs_params.rows = input_depth; in TransposeConvV2()
77 rhs_params.cols = input_image_size; in TransposeConvV2()
78 rhs_params.zero_point = -params.input_offset; in TransposeConvV2()
86 cpu_backend_gemm::Gemm(lhs_params, hwoi_ordered_filter_data, rhs_params, in TransposeConvV2()
Dconv.h102 cpu_backend_gemm::MatrixParams<int8> rhs_params; in ConvPerChannel() local
103 rhs_params.rows = gemm_input_rows; in ConvPerChannel()
104 rhs_params.cols = gemm_input_cols; in ConvPerChannel()
105 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in ConvPerChannel()
106 rhs_params.zero_point = -input_offset; in ConvPerChannel()
121 cpu_backend_gemm::Gemm(lhs_params, filter_data, rhs_params, gemm_input_data, in ConvPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dbatch_matmul.h85 MatrixParams<float> rhs_params; in BatchMatMul() local
86 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in BatchMatMul()
87 rhs_params.rows = accum_depth; in BatchMatMul()
88 rhs_params.cols = rhs_cols; in BatchMatMul()
108 cpu_backend_gemm::Gemm(lhs_params, lhs_ptr2, rhs_params, rhs_ptr2, in BatchMatMul()
195 MatrixParams<int8_t> rhs_params; in BatchMatMul() local
196 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in BatchMatMul()
197 rhs_params.rows = accum_depth; in BatchMatMul()
198 rhs_params.cols = rhs_cols; in BatchMatMul()
227 cpu_backend_gemm::Gemm(lhs_params, lhs_ptr2, rhs_params, rhs_ptr2, in BatchMatMul()
[all …]
Dsse_tensor_utils.cc190 MatrixParams<int8_t> rhs_params; in SseCpuBackendGemm() local
191 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in SseCpuBackendGemm()
192 rhs_params.rows = n_input; in SseCpuBackendGemm()
193 rhs_params.cols = n_batch; in SseCpuBackendGemm()
204 cpu_backend_gemm::Gemm(lhs_params, input_to_gate_weights, rhs_params, input, in SseCpuBackendGemm()
Doptimized_ops.h341 cpu_backend_gemm::MatrixParams<float> rhs_params; in FullyConnected() local
342 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in FullyConnected()
343 rhs_params.rows = input_rows; in FullyConnected()
344 rhs_params.cols = input_shape.FlatSize() / input_rows; in FullyConnected()
345 rhs_params.cache_policy = in FullyConnected()
347 TFLITE_DCHECK_EQ(input_shape.FlatSize(), rhs_params.rows * rhs_params.cols); in FullyConnected()
363 cpu_backend_gemm::Gemm(lhs_params, weights_data, rhs_params, input_data, in FullyConnected()
408 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in FullyConnected() local
409 rhs_params.rows = filter_cols; in FullyConnected()
410 rhs_params.cols = batches; in FullyConnected()
[all …]
Dneon_tensor_utils.cc993 MatrixParams<int8_t> rhs_params; in NeonCpuBackendGemm() local
994 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in NeonCpuBackendGemm()
995 rhs_params.rows = n_input; in NeonCpuBackendGemm()
996 rhs_params.cols = n_batch; in NeonCpuBackendGemm()
1007 cpu_backend_gemm::Gemm(lhs_params, input_to_gate_weights, rhs_params, input, in NeonCpuBackendGemm()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators.cc64 std::vector<XlaOp> rhs_params; in CreateScalarComparisonComputation() local
78 rhs_params.emplace_back(rhs_param); in CreateScalarComparisonComputation()
101 generators[i].value()(lhs_params[i], rhs_params[i], {}), in CreateScalarComparisonComputation()
105 And(param_equal, EqTotalOrder(lhs_params[i], rhs_params[i])); in CreateScalarComparisonComputation()