| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_partition_op_gpu.cu.cc | 198 Tensor* partitions_out, Tensor* indices_out, in AllocateTempSpace() argument
208 c, c->allocate_temp(DT_INT32, TensorShape({N}), indices_out), done); in AllocateTempSpace()
267 Tensor indices_out; in ComputeAsync() local
271 this->CountAndSortParts(c, &partitions, &partition_count, &indices_out, in ComputeAsync()
304 auto wrapped_callback = [this, c, &data, &partitions, indices_out, in ComputeAsync()
314 this->GatherSlices(c, &data, &indices_out, N, slice_size, outputs); in ComputeAsync()
326 Tensor* indices_out, DoneCallback done) { in RadixSort() argument
337 int32* indices_out_ptr = indices_out->flat<int32>().data(); in RadixSort()
359 Tensor* partition_count, Tensor* indices_out, in CountAndSortParts() argument
369 this->AllocateTempSpace(c, N, &indices_in, &partitions_out, indices_out, in CountAndSortParts()
[all …]
|
| D | sparse_cross_op.cc | 143 Tensor* indices_out, Tensor* values_out) in OutputUpdater() argument
145 indices_out_(indices_out), in OutputUpdater()
320 Tensor* indices_out; in Compute() local
325 CreateOutputTensors(columns, batch_size, context, &indices_out, &values_out, in Compute()
329 output_start_indices, indices_out, values_out); in Compute()
506 int64 batch_size, OpKernelContext* context, Tensor** indices_out, in CreateOutputTensors() argument
523 0, TensorShape({cross_count_total, 2}), indices_out)); in CreateOutputTensors()
|
| D | topk_op.cc | 81 Tensor* indices_out = nullptr; in Compute() local
83 context->allocate_output(1, output_shape, &indices_out)); in Compute()
89 auto indices = indices_out->flat_inner_dims<int32>(); in Compute()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_test.py | 631 indices_out, values_out, shape_out = sp_out
632 self.assertAllEqual(indices_out, indices)
641 indices_out, values_out, shape_out = s.run(sp)
642 self.assertAllEqual(indices_out, indices)
646 (indices_out, values_out, shape_out), = s.run([sp])
647 self.assertAllEqual(indices_out, indices)
656 indices_out, values_out, shape_out = s.run({'sp': sp})['sp']
657 self.assertAllEqual(indices_out, indices)
661 (indices_out, values_out, shape_out), = s.run({'sp': [sp]})['sp']
662 self.assertAllEqual(indices_out, indices)
[all …]
|
| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| D | sparse_feature_cross_op.py | 97 indices_out, values_out, shape_out = (
110 indices_out, values_out, shape_out = (
122 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
|
| /external/tensorflow/tensorflow/contrib/layers/kernels/ |
| D | sparse_feature_cross_kernel.cc | 143 Tensor* indices_out, Tensor* values_out) in OutputUpdater() argument
145 indices_out_(indices_out), in OutputUpdater()
363 Tensor* indices_out; in Compute() local
368 CreateOutputTensors(columns, batch_size, context, &indices_out, &values_out, in Compute()
372 updater(output_start_indices, indices_out, values_out); in Compute()
549 int64 batch_size, OpKernelContext* context, Tensor** indices_out, in CreateOutputTensors() argument
566 0, TensorShape({cross_count_total, 2}), indices_out)); in CreateOutputTensors()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2_indexed_slices_rewriter.py | 172 indices_out = grad_output_slices.indices
190 indices = constant_op.constant([], indices_out.dtype, name="indices_init")
|
| D | sparse_ops.py | 607 indices_out, values_out, shape_out = gen_sparse_ops.sparse_cross(
619 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
|