Searched defs:indices_out (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_partition_op_gpu.cu.cc | 205 Tensor* partitions_out, Tensor* indices_out, in AllocateTempSpace()
274 Tensor indices_out; in ComputeAsync() local
336 Tensor* indices_out, DoneCallback done) { in RadixSort()
369 Tensor* partition_count, Tensor* indices_out, in CountAndSortParts()
|
| D | sparse_cross_op.cc | 264 Tensor* indices_out, Tensor* values_out) in OutputUpdater()
662 int64 batch_size, OpKernelContext* context, Tensor** indices_out, in CreateOutputTensors()
724 Tensor* indices_out; in Compute() local
786 Tensor* indices_out; in Compute() local
855 Tensor* indices_out; in Compute() local
|
| D | topk_op.cc | 81 Tensor* indices_out = nullptr; in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 171 int64* indices_out, int size) { in COOMatrixToSparseTensorKernel2D()
194 const int* coo_rows, const int* coo_cols, int64* indices_out, in COOMatrixToSparseTensorKernel3D()
|