Home
last modified time | relevance | path

Searched defs:batch_mat_mul (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator_test.cc531 BatchMatMulDimensions* batch_mat_mul, in CountBatchMatMulOperations()
645 BatchMatMulDimensions batch_mat_mul; in CountBatchMatMulDimProduct() local
H A Dop_level_cost_estimator.cc1252 const OpInfo& op_info, BatchMatMulDimensions* batch_mat_mul, in CountBatchMatMulOperations()