Searched refs:dims_a (Results 1 – 1 of 1) sorted by relevance
111 OpContext DescribeBatchMatMul(const std::vector<int>& dims_a, in DescribeBatchMatMul() argument117 DescribeArbitraryRankInput(dims_a, DT_FLOAT, &op_context.op_info); in DescribeBatchMatMul()