Searched refs:input_itr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | permutation_input_iterator.h | 43 InputIteratorT input_itr;
49 InputIteratorT input_itr, ///< Input iterator to wrap in PermutationInputIterator() argument
51 : input_itr(input_itr), index_itr(index_itr) {} in PermutationInputIterator()
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);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | l2loss_op_gpu.cu.cc | 53 inputIterType input_itr((T*)input.flat<T>().data(), squareHalf<T>()); in Compute() local
58 context, (T*)output->flat<T>().data(), input_itr, 1, in Compute()
|
| D | softmax_op_gpu.cu.cc | 173 InputIterType input_itr( in Compute() local
181 input_itr, rows, cols); in Compute()
|
| D | reduction_gpu_kernels.cu.h | 920 inputIterType input_itr((T*)in.data(), Square<T>());
924 ctx, output_itr, input_itr, in.rank(), in.dimension(0),
995 inputIterType input_itr((Eigen::half*)in.data(), HalfToFloat());
1003 ctx, itr, input_itr, in.rank(), in.dimension(0),
|