Searched defs:batch_op (Results 1 – 2 of 2) sorted by relevance
34 const BatchMatMulOperator* batch_op, const std::vector<int> batch, in UnrollBatchMatMul3D()122 const BatchMatMulOperator* batch_op, Model* model, in UnrollBatchMatMulRecursion()221 const auto* batch_op = in Run() local
164 const string& batch_op, const string& map_input_name) { in CheckNotVectorized()