Searched defs:indices_out (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_partition_op_gpu.cu.cc | 198 Tensor* partitions_out, Tensor* indices_out, in AllocateTempSpace()
267 Tensor indices_out; in ComputeAsync() local
326 Tensor* indices_out, DoneCallback done) { in RadixSort()
359 Tensor* partition_count, Tensor* indices_out, in CountAndSortParts()
|
| D | sparse_cross_op.cc | 143 Tensor* indices_out, Tensor* values_out) in OutputUpdater()
320 Tensor* indices_out; in Compute() local
506 int64 batch_size, OpKernelContext* context, Tensor** indices_out, in CreateOutputTensors()
|
| D | topk_op.cc | 81 Tensor* indices_out = nullptr; in Compute() local
|
| /external/tensorflow/tensorflow/contrib/layers/kernels/ |
| D | sparse_feature_cross_kernel.cc | 143 Tensor* indices_out, Tensor* values_out) in OutputUpdater()
363 Tensor* indices_out; in Compute() local
549 int64 batch_size, OpKernelContext* context, Tensor** indices_out, in CreateOutputTensors()
|