Searched refs:pack_inputs (Results 1 – 1 of 1) sorted by relevance
36 std::vector<string>* pack_inputs) { in UnrollBatchMatMul3D() argument117 pack_inputs->push_back(matmul_op->outputs[0]); in UnrollBatchMatMul3D()143 std::vector<string> pack_inputs = UnrollBatchMatMulRecursion( in UnrollBatchMatMulRecursion() local150 pack_op->inputs = pack_inputs; in UnrollBatchMatMulRecursion()155 pack_op->values_count = pack_inputs.size(); in UnrollBatchMatMulRecursion()283 std::vector<string> pack_inputs = UnrollBatchMatMulRecursion( in Run() local286 pack_op->inputs = pack_inputs; in Run()289 pack_op->values_count = pack_inputs.size(); in Run()