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

  /external/tensorflow/tensorflow/core/kernels/
where_op.cc 95 typename TTypes<int64>::Matrix output, TIndex* found_true) {
111 if (FastBoundsCheck(*found_true, output_size)) {
112 WriteIndexRowMajor(output, strides, *found_true, n);
114 ++*found_true;
156 int64 found_true = 0; variable
162 output->matrix<int64>(), &found_true); \
181 context, found_true == num_true_t(),
186 found_true, " elements."));
228 typename TTypes<int64>::Matrix output, Tindex* found_true); \
310 // TODO(ebrevdo): Properly copy back found_true value to CPU fo
    [all...]
where_op.h 51 // found_true should sit on the host. Compute should copy the
53 // *found_true != output.dimension(0),
59 typename TTypes<int64>::Matrix output, TIndex* found_true);

Completed in 2698 milliseconds