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()
179 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_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
534 MatrixParams<RhsScalar> rhs_params; in TestMaybeValidGemm() local
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()
/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.cc282 MatrixParams<int8_t> rhs_params; in SseCpuBackendGemm() local
Doptimized_ops.h339 cpu_backend_gemm::MatrixParams<float> rhs_params; in FullyConnected() local
406 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in FullyConnected() local
467 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in FullyConnected() local
1313 cpu_backend_gemm::MatrixParams<float> rhs_params; in Conv() local
1498 cpu_backend_gemm::MatrixParams<int8> rhs_params; in HybridConvPerChannel() local
1617 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in Conv() local
2993 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in LstmCell() local
5320 cpu_backend_gemm::MatrixParams<float> rhs_params; in TransposeConvV2() local
5548 cpu_backend_gemm::MatrixParams<uint8_t> rhs_params; in TransposeConvV2() local
8172 cpu_backend_gemm::MatrixParams<float> rhs_params; in Conv3D() local
[all …]
Dneon_tensor_utils.cc993 MatrixParams<int8_t> rhs_params; in NeonCpuBackendGemm() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h69 cpu_backend_gemm::MatrixParams<int8> rhs_params; in FullyConnected() local
Dconv.h102 cpu_backend_gemm::MatrixParams<int8> rhs_params; in ConvPerChannel() local
Dtranspose_conv.h74 cpu_backend_gemm::MatrixParams<int8_t> rhs_params; in TransposeConvV2() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators.cc64 std::vector<XlaOp> rhs_params; in CreateScalarComparisonComputation() local