Searched refs:dims_b (Results 1 – 1 of 1) sorted by relevance
112 const std::vector<int>& dims_b) { in DescribeBatchMatMul() argument118 DescribeArbitraryRankInput(dims_b, DT_FLOAT, &op_context.op_info); in DescribeBatchMatMul()124 const std::vector<int>& dims_b, in DescribeSparseTensorDenseMatMul() argument133 DescribeArbitraryRankInput(dims_b, DT_FLOAT, &op_context.op_info); in DescribeSparseTensorDenseMatMul()