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.h235 static int64 CountMatMulOperations(const OpInfo& op_info,
237 static int64 CountMatMulOperations(const OpInfo& op_info,
Dop_level_cost_estimator_test.cc519 int64 CountMatMulOperations(const OpInfo& op_info, in CountMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimatorTest
521 return estimator_.CountMatMulOperations(op_info, found_unknown_shapes); in CountMatMulOperations()
1205 CountMatMulOperations(DescribeMatMul(2, 2, 4, 4).op_info, in TEST_P()
1210 EXPECT_EQ(10 * CountMatMulOperations(DescribeMatMul(2, 2, 4, 4).op_info, in TEST_P()
1216 EXPECT_EQ(20 * CountMatMulOperations(DescribeMatMul(2, 2, 4, 4).op_info, in TEST_P()
Dop_level_cost_estimator.cc1033 int64 OpLevelCostEstimator::CountMatMulOperations(const OpInfo& op_info, in CountMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator
1035 return CountMatMulOperations(op_info, nullptr, found_unknown_shapes); in CountMatMulOperations()
1039 int64 OpLevelCostEstimator::CountMatMulOperations(const OpInfo& op_info, in CountMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator
1356 ops += CountMatMulOperations(matmul_op_info, &matmul_unknown_shapes); in CountBatchMatMulOperations()
1796 int64 num_compute_ops = CountMatMulOperations(op_info, &found_unknown_shapes); in PredictMatMul()