Searched refs:k_array (Results 1 – 2 of 2) sorted by relevance
111 std::vector<MKL_INT> k_array(batch_size, K); in Compute() local129 &k_array[0], &a_array[0], &lda_array[0], &b_array[0], in Compute()
2197 string k_array = CreateConstArray<ArrayDataType::kInt32>( in ConvertTopKV2Operator() local2199 op->inputs.push_back(k_array); in ConvertTopKV2Operator()