HomeSort by relevance Sort by last modified time
    Searched refs:TopK (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
topk_op.cc 40 class TopK : public OpKernel {
42 explicit TopK(OpKernelConstruction* context) : OpKernel(context) {
44 if (num_inputs() < 2) { // k is an attr (TopK).
235 TopK<CPUDevice, type>)
238 REGISTER_KERNELS_NAME(TopK, type); \
266 Name("TopK").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
267 TopK<GPUDevice, type>) \
272 TopK<GPUDevice, type>)
nn_ops_test.cc     [all...]
  /external/tensorflow/tensorflow/contrib/lite/kernels/
topk_v2.cc 45 "TopK k input must have 1 or more dimensions.");
48 "TopK k is higher than the internal dimension.");
137 void TopK(int32 row_size, int32 num_rows, const T* data, int32 k,
202 TopK(row_size, num_rows, input->data.f, k, output_indexes->data.i32,
206 TopK(row_size, num_rows, input->data.uint8, k, output_indexes->data.i32,
210 TopK(row_size, num_rows, input->data.i32, k, output_indexes->data.i32,
214 TopK(row_size, num_rows, input->data.i64, k, output_indexes->data.i32,
218 context->ReportError(context, "Type is currently not supported by TopK.");
  /external/tensorflow/tensorflow/examples/label_image/
main.cc 210 TopK(root.WithOpName(output_name), outputs[0], how_many_labels);
219 // The TopK node returns two outputs, the scores and their original indices,
  /external/tensorflow/tensorflow/examples/multibox_detector/
main.cc 187 TopK(root.WithOpName(output_name), outputs[0], how_many_labels);
196 // The TopK node returns two outputs, the scores and their original indices,
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 162 milliseconds