| /external/tensorflow/tensorflow/core/util/ |
| permutation_input_iterator.h | 43 InputIteratorT input_itr; member in class:tensorflow::PermutationInputIterator
49 InputIteratorT input_itr, ///< Input iterator to wrap
51 : input_itr(input_itr), index_itr(index_itr) {}
68 return input_itr[*index_itr];
74 self_type retval(input_itr, index_itr + n);
88 self_type retval(input_itr, index_itr - n);
108 return input_itr[index_itr[n]];
113 return input_itr + *index_itr;
118 return (index_itr == rhs.index_itr && input_itr == rhs.input_itr)
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| l2loss_op_gpu.cu.cc | 53 inputIterType input_itr((T*)input.flat<T>().data(), squareHalf<T>());
58 context, (T*)output->flat<T>().data(), input_itr, 1,
|
| softmax_op_gpu.cu.cc | 127 InputIterType input_itr(
134 context, const_cast<T*>(sum_probs.flat<T>().data()), input_itr, rows,
|
| reduction_gpu_kernels.cu.h | 900 inputIterType input_itr((Eigen::half*)in.data(), HalfToFloat());
908 ctx, itr, input_itr, in.rank(), in.dimension(0),
[all...] |
