Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h155 static std::unique_ptr<Array2D<Eigen::half>> MatmulArray2D(
157 static std::unique_ptr<Array2D<float>> MatmulArray2D(
159 static std::unique_ptr<Array2D<double>> MatmulArray2D(
353 std::unique_ptr<Array2D<float>> MatmulArray2D(const Array2D<float>& lhs,
Dhlo_evaluator.cc1638 std::unique_ptr<Array2D<Eigen::half>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
1644 std::unique_ptr<Array2D<float>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
1650 std::unique_ptr<Array2D<double>> HloEvaluator::MatmulArray2D( in MatmulArray2D() function in xla::HloEvaluator
Dhlo_evaluator_typed_visitor.h1260 HloEvaluator::MatmulArray2D(lhs_array, rhs_array); in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc35 /* static */ std::unique_ptr<Array2D<Eigen::half>> ReferenceUtil::MatmulArray2D( in MatmulArray2D() function in xla::ReferenceUtil
37 return HloEvaluator::MatmulArray2D(lhs, rhs); in MatmulArray2D()
40 /* static */ std::unique_ptr<Array2D<float>> ReferenceUtil::MatmulArray2D( in MatmulArray2D() function in xla::ReferenceUtil
42 return HloEvaluator::MatmulArray2D(lhs, rhs); in MatmulArray2D()
45 /* static */ std::unique_ptr<Array2D<double>> ReferenceUtil::MatmulArray2D( in MatmulArray2D() function in xla::ReferenceUtil
47 return HloEvaluator::MatmulArray2D(lhs, rhs); in MatmulArray2D()
Dreference_util_test.cc61 TEST_F(ReferenceUtilTest, MatmulArray2D) { in TEST_F() argument
67 auto result = ReferenceUtil::MatmulArray2D(*matrix_, rhs); in TEST_F()
Dreference_util.h57 static std::unique_ptr<Array2D<Eigen::half>> MatmulArray2D(
59 static std::unique_ptr<Array2D<float>> MatmulArray2D(
61 static std::unique_ptr<Array2D<double>> MatmulArray2D(
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddot_operation_test.cc345 *ReferenceUtil::MatmulArray2D(*dot_lhs_data, *dot_rhs_data), in TestImpl()
348 expected = ReferenceUtil::MatmulArray2D(*dot_lhs_data, *dot_rhs_data); in TestImpl()