Searched defs:indices_out (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_partition_op_gpu.cu.cc | 194 Tensor* partitions_out, Tensor* indices_out, in AllocateTempSpace()
263 Tensor indices_out; in ComputeAsync() local
325 Tensor* indices_out, DoneCallback done) { in RadixSort()
358 Tensor* partition_count, Tensor* indices_out, in CountAndSortParts()
|
| D | sparse_cross_op.cc | 278 Tensor* indices_out, Tensor* values_out) in OutputUpdater()
710 int64_t batch_size, OpKernelContext* context, Tensor** indices_out, in CreateOutputTensors()
775 Tensor* indices_out; in Compute() local
846 Tensor* indices_out; in Compute() local
919 Tensor* indices_out; in Compute() local
|
| D | topk_op.cc | 91 Tensor* indices_out = nullptr; in Compute() local
|
| D | gpu_prim_helpers_test.cu.cc | 63 Tensor* indices_out; 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()
|