Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm.h117 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in Gemm()
177 const MatrixParams<int16_t>& rhs_params, const int16_t* rhs_data, in Gemm()
202 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()
413 MatrixParams<RhsScalar> rhs_params; in TestSomeGemm() local
554 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/integer_ops/
Dfully_connected.h73 cpu_backend_gemm::MatrixParams<InputScalar> rhs_params; in FullyConnectedPerChannel() local
145 cpu_backend_gemm::MatrixParams<InputScalar> rhs_params; in FullyConnected() local
Dtranspose_conv.h79 cpu_backend_gemm::MatrixParams<InputScalar> rhs_params; in TransposeConvV2() local
Dconv.h103 cpu_backend_gemm::MatrixParams<InputScalar> 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
446 MatrixParams<int8_t> rhs_params; in BatchMatMul() local
Dsse_tensor_utils.cc392 MatrixParams<int8_t> rhs_params; in SseCpuBackendGemm() local
Doptimized_ops.h277 cpu_backend_gemm::MatrixParams<float> rhs_params; in FullyConnected() local
344 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in FullyConnected() local
405 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in FullyConnected() local
984 cpu_backend_gemm::MatrixParams<float> rhs_params; in Conv() local
1169 cpu_backend_gemm::MatrixParams<int8> rhs_params; in HybridConvPerChannel() local
1288 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in Conv() local
2668 cpu_backend_gemm::MatrixParams<uint8> rhs_params; in LstmCell() local
4997 cpu_backend_gemm::MatrixParams<float> rhs_params; in TransposeConvV2() local
5498 cpu_backend_gemm::MatrixParams<uint8_t> rhs_params; in TransposeConvV2() local
8122 cpu_backend_gemm::MatrixParams<float> rhs_params; in Conv3D() local
[all …]
Dneon_tensor_utils.cc1019 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
Dapprox_topk.cc75 std::vector<XlaOp> rhs_params; in BuildReductionComputation() local