OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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