Searched refs:CountBatchMatMulDimProduct (Results 1 – 1 of 1) sorted by relevance
609 int64 CountBatchMatMulDimProduct(const OpInfo& op_info, in CountBatchMatMulDimProduct() function in tensorflow::grappler::OpLevelBatchMatMulCostEstimatorTest1224 int prod = CountBatchMatMulDimProduct( in TEST_P()1234 CountBatchMatMulDimProduct(bad_batch.op_info, &batch_matmul_inaccurate); in TEST_P()1245 prod = CountBatchMatMulDimProduct(transpose_batch.op_info, in TEST_P()