Searched refs:input_k (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dtranspose.cc102 for (int output_k = 0, input_k = size - 1; output_k < size;
103 ++output_k, --input_k) {
104 reversed_perm[output_k] = size - perm_data[input_k] - 1;
H A Dtranspose_test.cc44 for (int output_k = 0, input_k = shape.size() - 1; output_k < shape.size();
45 output_k++, input_k--) {
46 reversed_perms[output_k] = shape.size() - perms[input_k] - 1;
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc987 const auto& input_k = model->GetArray(op->inputs[1]); local
1011 if (input_k.buffer) {
1012 const int32_t k_value = input_k.GetBuffer<ArrayDataType::kInt32>().data[0];
/external/tensorflow/tensorflow/core/kernels/
H A Dnn_ops_test.cc1310 Tensor input_k(DT_INT32, TensorShape({}));
1311 input_k.scalar<int32>()() = k;
1313 auto top_k = ops::TopK(root, input, input_k, ops::TopK::Sorted(true));

Completed in 173 milliseconds