Home
last modified time | relevance | path

Searched defs:rhs_data (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/ruy/example/
H A Dexample.cc23 const float rhs_data[] = {1, 2, 3, 4}; in ExampleMulFloat() local
47 const float rhs_data[] = {1, 2, 3, 4}; in ExampleMulFloatWithBiasAddAndClamp() local
75 const std::uint8_t rhs_data[] = {129, 130, 131, 132}; in ExampleMulUint8AsymmetricQuantized() local
104 const std::int8_t rhs_data[] = {1, 2, 3, 4}; in ExampleMulInt8PerChannelQuantized() local
132 const std::int8_t rhs_data[] = {1, 2, 3, 4}; in ExampleMulInt8GetRawAccumulators() local
157 const std::int16_t rhs_data[] = {1000, 2000, 3000, 4000}; in ExampleMulInt8TimesInt16PerChannelQuantized() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A 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()
H A Dcpu_backend_gemm_test.cc256 const std::vector<RhsScalar>& rhs_data, in PerformGemmThenCompareResultsThenAgainWithClamping()
314 const std::vector<RhsScalar>& rhs_data, in BisectReasonableMultiplierExponent()
349 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data, in ReferenceGemm()
378 std::vector<RhsScalar> rhs_data; in TestSomeGemm() local
533 std::vector<RhsScalar> rhs_data; in TestMaybeValidGemm() local
H A 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()
H A Dcpu_backend_gemm_eigen.cc39 const MatrixParams<float>& rhs_params, const float* rhs_data, in Run()
H A Dbatch_matmul_test.cc325 std::initializer_list<float> rhs_data, in ConstRHSBatchMatMulOpModel()
H A Dbatch_matmul.cc308 const TfLiteTensor* rhs_data; in Prepare() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dtridiagonal_test.cc59 auto rhs_data = CreateR3Parameter<float>(rhs, 3, "rhs", &builder, &rhs_xla); in XLA_TEST_P() local
99 auto rhs_data = CreateParameter<float>(rhs, 3, "rhs", &builder, &rhs_xla); in XLA_TEST_P() local
140 auto rhs_data = CreateR3Parameter<float>(rhs, 3, "rhs", &builder, &rhs_xla); in XLA_TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dbatch_matmul.h29 const RuntimeShape& rhs_shape, const float* rhs_data, in BatchMatMul()
116 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
273 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
379 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dbatch_matmul.h55 const RuntimeShape& rhs_shape, const Tb* rhs_data, in BatchMatMul()
110 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
202 const RuntimeShape& rhs_shape, const T* rhs_data, in BatchMatMul()
/aosp_15_r20/external/ruy/ruy/
H A Dprepacked_cache_test.cc236 const float rhs_data[] = {1, 2}; in TestCachePolicies() local
277 const float rhs_data[] = {1, 2}; in TEST() local
H A Dtest_overflow_dst_zero_point.cc62 const std::vector<std::int8_t> rhs_data(cols, 0); in TestOverflowingAdditionOfDestinationZeroPoint() local
H A Dperchannel_buffers_reallocation_test.cc60 const RhsScalar rhs_data[kMatrixSize * kMatrixSize] = {0}; in TestPerChannelBuffersReallocation() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_op_test.cc100 void RunMatMulWithBias(const Tensor& lhs_data, const Tensor& rhs_data, in RunMatMulWithBias()
120 const Tensor& lhs_data, const Tensor& rhs_data, const Tensor& bias_data, in RunMatMulWithBiasAndActivation()
150 void RunFusedMatMulOp(const Tensor& lhs_data, const Tensor& rhs_data, in RunFusedMatMulOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Duniform_quantized_dot_ops.cc66 const Trhs* rhs_data = rhs.flat<Trhs>().data(); in DotWithAccFunctionAndOutputFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dtridiagonal_solve_op_gpu.cu.cc285 const Scalar* rhs_data = rhs.flat<Scalar>().data(); in ComputeWithGtsvBatched() local
/aosp_15_r20/external/angle/third_party/abseil-cpp/absl/types/
H A Dspan_test.cc834 static constexpr int rhs_data[] = {1, 2, 3}; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dtooling_util.cc781 const auto& rhs_data = rhs_array.GetBuffer<A>().data; in CompareArrayBuffers() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Darray_elementwise_ops_test.cc281 std::unique_ptr<GlobalData> rhs_data = in XLA_TEST_F() local
320 std::unique_ptr<GlobalData> rhs_data = in XLA_TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.cc523 se::DeviceMemoryBase rhs_data = GetDeviceAddress(rhs); in operator ()() local
/aosp_15_r20/external/mesa3d/src/imgui/
H A Dimgui.h1208 …rhs_cap = rhs.Capacity; rhs.Capacity = Capacity; Capacity = rhs_cap; T* rhs_data = rhs.Data; rhs.D… in swap() local