Searched refs:logits_t (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | multinomial_op.cc | 153 void DoCompute(OpKernelContext* ctx, const Tensor& logits_t, in DoCompute() argument
155 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(logits_t.shape()), in DoCompute()
157 logits_t.shape().DebugString())); in DoCompute()
169 const int64 dim = logits_t.dim_size(i); in DoCompute()
172 "logits.shape = ", logits_t.shape().DebugString(), in DoCompute()
175 const int batch_size = static_cast<int>(logits_t.dim_size(0)); in DoCompute()
176 const int num_classes = static_cast<int>(logits_t.dim_size(1)); in DoCompute()
212 ctx, ctx->eigen_device<Device>(), logits_t.matrix<T>(), in DoCompute()
229 const Tensor& logits_t = ctx->input(0); in Compute() local
231 this->DoCompute(ctx, logits_t, num_samples_t, &generator_); in Compute()
[all …]
|
| D | multinomial_op_test.cc | 29 Tensor logits_t(DT_FLOAT, TensorShape({batch_size, num_classes})); in Multinomial() local
31 logits_t.flat<float>().setRandom(); in Multinomial()
36 .Input(test::graph::Constant(g, logits_t)) in Multinomial()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 1040 def compute_ctc_loss(logits_t, labels_t, label_length_t, logit_length_t, argument
1043 logits_t.set_shape(logits.shape)
1048 logits=logits_t,
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 926 logits_t = math_ops.cast(logits, dtype)
929 sparse_logits_t = dense_to_csr_sparse_matrix(logits_t)
950 logits_t = math_ops.cast(logits, dtype)
953 sparse_logits_t = dense_to_csr_sparse_matrix(logits_t)
|