Home
last modified time | relevance | path

Searched refs:MatmulArray2D (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator.h218 static std::unique_ptr<Array2D<Eigen::half>> MatmulArray2D(
220 static std::unique_ptr<Array2D<float>> MatmulArray2D(
222 static std::unique_ptr<Array2D<double>> MatmulArray2D(
224 static std::unique_ptr<Array2D<std::complex<float>>> MatmulArray2D(
227 static std::unique_ptr<Array2D<std::complex<double>>> MatmulArray2D(
230 static std::unique_ptr<Array2D<int32_t>> MatmulArray2D(
479 std::unique_ptr<Array2D<float>> MatmulArray2D(const Array2D<float>& lhs,
H A Dtriangular_solve_expander_test.cc92 auto ref_b = ReferenceUtil::MatmulArray2D(a, x); in TEST_P()
H A Dhlo_evaluator.cc3901 std::unique_ptr<Array2D<Eigen::half>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
3907 std::unique_ptr<Array2D<float>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
3913 std::unique_ptr<Array2D<double>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
3919 std::unique_ptr<Array2D<std::complex<float>>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
3926 std::unique_ptr<Array2D<std::complex<double>>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
3933 std::unique_ptr<Array2D<int32_t>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
H A Dhlo_evaluator_typed_visitor.h1210 HloEvaluator::MatmulArray2D(lhs_array, rhs_array);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util_test.cc60 TEST_F(ReferenceUtilTest, MatmulArray2D) { in TEST_F() argument
66 auto result = ReferenceUtil::MatmulArray2D(*matrix_, rhs); in TEST_F()
H A Dreference_util.h56 static std::unique_ptr<Array2D<T>> MatmulArray2D(const Array2D<T>& lhs, in MatmulArray2D() function
58 return HloEvaluator::MatmulArray2D(lhs, rhs); in MatmulArray2D()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Ddot_operation_test.cc379 *ReferenceUtil::MatmulArray2D(*dot_lhs_data, *dot_rhs_data), in TestImpl()
382 expected = ReferenceUtil::MatmulArray2D(*dot_lhs_data, *dot_rhs_data); in TestImpl()