Home
last modified time | relevance | path

Searched defs:lhs_params (Results 1 – 12 of 12) 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()
178 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
526 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/
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
Dsse_tensor_utils.cc276 MatrixParams<int8_t> lhs_params; in SseCpuBackendGemm() local
Doptimized_ops.h346 cpu_backend_gemm::MatrixParams<float> lhs_params; in FullyConnected() local
399 cpu_backend_gemm::MatrixParams<uint8> lhs_params; in FullyConnected() local
460 cpu_backend_gemm::MatrixParams<uint8> lhs_params; in FullyConnected() local
1309 cpu_backend_gemm::MatrixParams<float> lhs_params; in Conv() local
1493 cpu_backend_gemm::MatrixParams<int8> lhs_params; in HybridConvPerChannel() local
1612 cpu_backend_gemm::MatrixParams<uint8> lhs_params; in Conv() local
2988 cpu_backend_gemm::MatrixParams<uint8> lhs_params; in LstmCell() local
5313 cpu_backend_gemm::MatrixParams<float> lhs_params; in TransposeConvV2() local
5539 cpu_backend_gemm::MatrixParams<uint8_t> lhs_params; in TransposeConvV2() local
8168 cpu_backend_gemm::MatrixParams<float> lhs_params; in Conv3D() local
[all …]
Dneon_tensor_utils.cc987 MatrixParams<int8_t> lhs_params; in NeonCpuBackendGemm() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h64 cpu_backend_gemm::MatrixParams<int8> lhs_params; in FullyConnected() local
Dconv.h97 cpu_backend_gemm::MatrixParams<int8> lhs_params; in ConvPerChannel() local
Dtranspose_conv.h64 cpu_backend_gemm::MatrixParams<int8_t> lhs_params; in TransposeConvV2() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators.cc63 std::vector<XlaOp> lhs_params; in CreateScalarComparisonComputation() local