Home
last modified time | relevance | path

Searched refs:CountMatMulOperations (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.h83 int64 CountMatMulOperations(const OpInfo& op_info,
85 int64 CountMatMulOperations(const OpInfo& op_info, MatMulDimensions* mat_mul,
Dop_level_cost_estimator_test.cc457 int64 CountMatMulOperations(const OpInfo& op_info, in CountMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimatorTest
459 return estimator_.CountMatMulOperations(op_info, found_unknown_shapes); in CountMatMulOperations()
854 CountMatMulOperations(DescribeMatMul(2, 2, 4, 4).op_info, in TEST_F()
859 EXPECT_EQ(10 * CountMatMulOperations(DescribeMatMul(2, 2, 4, 4).op_info, in TEST_F()
865 EXPECT_EQ(20 * CountMatMulOperations(DescribeMatMul(2, 2, 4, 4).op_info, in TEST_F()
Dop_level_cost_estimator.cc680 int64 OpLevelCostEstimator::CountMatMulOperations( in CountMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator
682 return CountMatMulOperations(op_info, nullptr, found_unknown_shapes); in CountMatMulOperations()
686 int64 OpLevelCostEstimator::CountMatMulOperations( in CountMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator
859 ops += CountMatMulOperations(matmul_op_info, &matmul_unknown_shapes); in CountBatchMatMulOperations()
1228 CountMatMulOperations(op_info, &found_unknown_shapes), op_info); in PredictMatMul()