Searched refs:MatmulArray2D (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.h | 161 static std::unique_ptr<Array2D<Eigen::half>> MatmulArray2D(
163 static std::unique_ptr<Array2D<float>> MatmulArray2D(
165 static std::unique_ptr<Array2D<double>> MatmulArray2D(
167 static std::unique_ptr<Array2D<std::complex<float>>> MatmulArray2D(
170 static std::unique_ptr<Array2D<std::complex<double>>> MatmulArray2D(
173 static std::unique_ptr<Array2D<int32>> MatmulArray2D(
377 std::unique_ptr<Array2D<float>> MatmulArray2D(const Array2D<float>& lhs,
|
| D | triangular_solve_expander_test.cc | 92 auto ref_b = ReferenceUtil::MatmulArray2D(a, x); in TEST_P()
|
| D | hlo_evaluator.cc | 2578 std::unique_ptr<Array2D<Eigen::half>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
2584 std::unique_ptr<Array2D<float>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
2590 std::unique_ptr<Array2D<double>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
2596 std::unique_ptr<Array2D<std::complex<float>>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
2603 std::unique_ptr<Array2D<std::complex<double>>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
2610 std::unique_ptr<Array2D<int32>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
|
| D | hlo_evaluator_typed_visitor.h | 1380 HloEvaluator::MatmulArray2D(lhs_array, rhs_array);
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util_test.cc | 61 TEST_F(ReferenceUtilTest, MatmulArray2D) { in TEST_F() argument
67 auto result = ReferenceUtil::MatmulArray2D(*matrix_, rhs); in TEST_F()
|
| D | reference_util.h | 59 static std::unique_ptr<Array2D<T>> MatmulArray2D(const Array2D<T>& lhs, in MatmulArray2D() function
61 return HloEvaluator::MatmulArray2D(lhs, rhs); in MatmulArray2D()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 380 *ReferenceUtil::MatmulArray2D(*dot_lhs_data, *dot_rhs_data), in TestImpl()
383 expected = ReferenceUtil::MatmulArray2D(*dot_lhs_data, *dot_rhs_data); in TestImpl()
|