| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| 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/core/kernels/ |
| dynamic_partition_op_gpu.cu.cc | 21 // partitions_out and indices_out, with partitions_out sorted.
30 // 5. Finally, we use indices_out and the gather functor to collect the output.
31 // This works, because for each interval of i-values, indices_out points
197 Tensor* partitions_out, Tensor* indices_out,
207 c, c->allocate_temp(DT_INT32, TensorShape({N}), indices_out), done);
266 Tensor indices_out; local
268 // Also sort the info in partitions and output it in indices_out,
270 this->CountAndSortParts(c, &partitions, &partition_count, &indices_out,
303 auto wrapped_callback = [this, c, &data, &partitions, indices_out,
313 this->GatherSlices(c, &data, &indices_out, N, slice_size, outputs)
[all...] |
| sparse_cross_op.cc | 143 Tensor* indices_out, Tensor* values_out)
145 indices_out_(indices_out),
320 Tensor* indices_out; variable
325 CreateOutputTensors(columns, batch_size, context, &indices_out, &values_out,
329 output_start_indices, indices_out, values_out);
506 int64 batch_size, OpKernelContext* context, Tensor** indices_out,
523 0, TensorShape({cross_count_total, 2}), indices_out));
|
| topk_op.cc | 81 Tensor* indices_out = nullptr; variable
83 context->allocate_output(1, output_shape, &indices_out));
89 auto indices = indices_out->flat_inner_dims<int32>();
|
| /external/tensorflow/tensorflow/python/client/ |
| session_test.py | 546 indices_out, values_out, shape_out = sp_out
547 self.assertAllEqual(indices_out, indices)
556 indices_out, values_out, shape_out = s.run(sp)
557 self.assertAllEqual(indices_out, indices)
561 (indices_out, values_out, shape_out), = s.run([sp])
562 self.assertAllEqual(indices_out, indices)
571 indices_out, values_out, shape_out = s.run({'sp': sp})['sp']
572 self.assertAllEqual(indices_out, indices)
576 (indices_out, values_out, shape_out), = s.run({'sp': [sp]})['sp']
577 self.assertAllEqual(indices_out, indices
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/kernels/ |
| sparse_feature_cross_kernel.cc | 143 Tensor* indices_out, Tensor* values_out)
145 indices_out_(indices_out),
363 Tensor* indices_out; variable
368 CreateOutputTensors(columns, batch_size, context, &indices_out, &values_out,
372 updater(output_start_indices, indices_out, values_out);
549 int64 batch_size, OpKernelContext* context, Tensor** indices_out,
566 0, TensorShape({cross_count_total, 2}), indices_out));
|
| /external/tensorflow/tensorflow/python/ops/ |
| sparse_ops.py | 435 indices_out, values_out, shape_out = gen_sparse_ops._sparse_cross(
447 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
[all...] |
