Home
last modified time | relevance | path

Searched defs:input_k (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_matrix_set_diag_v2_or_v3_to_v1.cc47 const auto& input_k = model->GetArray(op->inputs[2]); in Run() local
Dconvert_matrix_diag_v2_or_v3_to_v1.cc41 const auto& input_k = model->GetArray(op->inputs[1]); in Run() local
Dpropagate_fixed_sizes.cc1467 const auto& input_k = model->GetArray(op->inputs[1]); in ProcessTopkV2Operator() local
/external/pytorch/aten/src/ATen/native/cuda/
DMultiLabelMarginCriterion.cu68 const scalar_t* input_k = input + k * dim; in C10_LAUNCH_BOUNDS_1() local
141 const scalar_t* input_k = input + k * dim; in C10_LAUNCH_BOUNDS_1() local
DMultiMarginLoss.cu30 const scalar_t *input_k = input + k*dim; in MultiMarginLoss_forward_kernel() local
76 const scalar_t *input_k = input + k*dim; in MultiMarginLoss_backward_kernel() local
/external/tensorflow/tensorflow/core/kernels/
Dnn_ops_test.cc1443 Tensor input_k(DT_INT32, TensorShape({})); in BM_TopK() local
/external/pytorch/torch/csrc/jit/frontend/
Dir_emitter.cpp1812 NamedValue input_k = NamedValue(loc, "", k); in emitDictComprehension() local
3857 NamedValue input_k = NamedValue(kwarg.range(), "", k); in emitApplySpecialFormForDict() local