Home
last modified time | relevance | path

Searched defs:lhs_params (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm.h116 void Gemm(const MatrixParams<LhsScalar>& lhs_params, const LhsScalar* lhs_data, in Gemm()
176 void Gemm(const MatrixParams<int8_t>& lhs_params, const int8_t* lhs_data, in Gemm()
201 void Gemm(const MatrixParams<LhsScalar>& lhs_params, const LhsScalar* lhs_data, in Gemm()
Dcpu_backend_gemm_eigen.cc38 const MatrixParams<float>& lhs_params, const float* lhs_data, in Run()
Dcpu_backend_gemm_test.cc253 const MatrixParams<LhsScalar>& lhs_params, in PerformGemmThenCompareResultsThenAgainWithClamping()
311 int bisect_min, int bisect_max, const MatrixParams<LhsScalar>& lhs_params, in BisectReasonableMultiplierExponent()
348 const MatrixParams<LhsScalar>& lhs_params, const LhsScalar* lhs_data, in ReferenceGemm()
396 MatrixParams<LhsScalar> lhs_params; in TestSomeGemm() local
541 MatrixParams<LhsScalar> lhs_params; in TestMaybeValidGemm() local
Dcpu_backend_gemm_custom_gemv.h102 const MatrixParams<LhsScalar>& lhs_params, const LhsScalar* lhs_data, in CustomGemvTask()
148 const MatrixParams<LhsScalar>& lhs_params, const LhsScalar* lhs_data, in CustomGemv()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h65 cpu_backend_gemm::MatrixParams<int8> lhs_params; in FullyConnectedPerChannel() local
137 cpu_backend_gemm::MatrixParams<int8> lhs_params; in FullyConnected() local
Dtranspose_conv.h69 cpu_backend_gemm::MatrixParams<int8_t> lhs_params; in TransposeConvV2() local
Dconv.h98 cpu_backend_gemm::MatrixParams<int8> lhs_params; in ConvPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dbatch_matmul.h80 MatrixParams<float> lhs_params; in BatchMatMul() local
190 MatrixParams<int8_t> lhs_params; in BatchMatMul() local
334 MatrixParams<int8_t> lhs_params; in BatchMatMul() local
440 MatrixParams<int8_t> lhs_params; in BatchMatMul() local
Dsse_tensor_utils.cc386 MatrixParams<int8_t> lhs_params; in SseCpuBackendGemm() local
Doptimized_ops.h284 cpu_backend_gemm::MatrixParams<float> lhs_params; in FullyConnected() local
337 cpu_backend_gemm::MatrixParams<uint8> lhs_params; in FullyConnected() local
398 cpu_backend_gemm::MatrixParams<uint8> lhs_params; in FullyConnected() local
980 cpu_backend_gemm::MatrixParams<float> lhs_params; in Conv() local
1164 cpu_backend_gemm::MatrixParams<int8> lhs_params; in HybridConvPerChannel() local
1283 cpu_backend_gemm::MatrixParams<uint8> lhs_params; in Conv() local
2663 cpu_backend_gemm::MatrixParams<uint8> lhs_params; in LstmCell() local
4990 cpu_backend_gemm::MatrixParams<float> lhs_params; in TransposeConvV2() local
5489 cpu_backend_gemm::MatrixParams<uint8_t> lhs_params; in TransposeConvV2() local
8118 cpu_backend_gemm::MatrixParams<float> lhs_params; in Conv3D() local
[all …]
Dneon_tensor_utils.cc1013 MatrixParams<int8_t> lhs_params; in NeonCpuBackendGemm() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators.cc63 std::vector<XlaOp> lhs_params; in CreateScalarComparisonComputation() local
Dapprox_topk.cc74 std::vector<XlaOp> lhs_params; in BuildReductionComputation() local