Home
last modified time | relevance | path

Searched defs:rhs_params (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm.h117 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in Gemm()
172 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in Gemm()
Dcpu_backend_gemm_eigen.cc39 const MatrixParams<float>& rhs_params, const float* rhs_data, in Run()
Dcpu_backend_gemm_custom_gemv.h103 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in CustomGemvTask()
149 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in CustomGemv()
Dcpu_backend_gemm_test.cc255 const MatrixParams<RhsScalar>& rhs_params, in PerformGemmThenCompareResultsThenAgainWithClamping()
313 const MatrixParams<RhsScalar>& rhs_params, in BisectReasonableMultiplierExponent()
349 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in ReferenceGemm()
408 MatrixParams<RhsScalar> rhs_params; in TestSomeGemm() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h69 cpu_backend_gemm::MatrixParams<int8> rhs_params; in FullyConnected() local
Dtranspose_conv.h74 cpu_backend_gemm::MatrixParams<int8_t> rhs_params; in TransposeConvV2() local
Dconv.h102 cpu_backend_gemm::MatrixParams<int8> rhs_params; in ConvPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dbatch_matmul.h85 MatrixParams<float> rhs_params; in BatchMatMul() local
195 MatrixParams<int8_t> rhs_params; in BatchMatMul() local
340 MatrixParams<int8_t> rhs_params; in BatchMatMul() local
Dsse_tensor_utils.cc190 MatrixParams<int8_t> rhs_params; in SseCpuBackendGemm() local
Doptimized_ops.h341 cpu_backend_gemm::MatrixParams<float> rhs_params; in FullyConnected() local
408 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in FullyConnected() local
469 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in FullyConnected() local
1315 cpu_backend_gemm::MatrixParams<float> rhs_params; in Conv() local
1500 cpu_backend_gemm::MatrixParams<int8> rhs_params; in HybridConvPerChannel() local
1619 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in Conv() local
2991 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in LstmCell() local
5758 cpu_backend_gemm::MatrixParams<float> rhs_params; in TransposeConvV2() local
5986 cpu_backend_gemm::MatrixParams<uint8_t> rhs_params; in TransposeConvV2() local
Dneon_tensor_utils.cc993 MatrixParams<int8_t> rhs_params; in NeonCpuBackendGemm() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators.cc64 std::vector<XlaOp> rhs_params; in CreateScalarComparisonComputation() local