Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.h210 struct BatchMatMulDimensions { struct
246 BatchMatMulDimensions* batch_mat_mul,
Dop_level_cost_estimator_test.cc513 using BatchMatMulDimensions = OpLevelCostEstimator::BatchMatMulDimensions; typedef in tensorflow::grappler::OpLevelCostEstimatorTest
530 BatchMatMulDimensions* batch_mat_mul, in CountBatchMatMulOperations()
611 BatchMatMulDimensions batch_mat_mul; in CountBatchMatMulDimProduct()
Dop_level_cost_estimator.cc1240 const OpInfo& op_info, BatchMatMulDimensions* batch_mat_mul, in CountBatchMatMulOperations()