Searched defs:input_k (Results 1 – 8 of 8) sorted by relevance
68 const scalar_t* input_k = input + k * dim; in C10_LAUNCH_BOUNDS_1() local141 const scalar_t* input_k = input + k * dim; in C10_LAUNCH_BOUNDS_1() local
30 const scalar_t *input_k = input + k*dim; in MultiMarginLoss_forward_kernel() local76 const scalar_t *input_k = input + k*dim; in MultiMarginLoss_backward_kernel() local
47 const auto& input_k = model->GetArray(op->inputs[2]); in Run() local
41 const auto& input_k = model->GetArray(op->inputs[1]); in Run() local
1467 const auto& input_k = model->GetArray(op->inputs[1]); in ProcessTopkV2Operator() local
60 for (int32_t output_k = 0, input_k = permSize - 1; output_k < permSize; ++output_k, --input_k) { in transposeGeneric() local
1443 Tensor input_k(DT_INT32, TensorShape({})); in BM_TopK() local
1812 NamedValue input_k = NamedValue(loc, "", k); in emitDictComprehension() local3857 NamedValue input_k = NamedValue(kwarg.range(), "", k); in emitApplySpecialFormForDict() local