Home
last modified time | relevance | path

Searched refs:logits_shape (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsoftmax_op.cc40 const TensorShape logits_shape = ctx->InputShape(0); in Compile() local
41 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(logits_shape), in Compile()
43 logits_shape.DebugString())); in Compile()
47 std::vector<int64> batch_dims(logits_shape.dims() - 1); in Compile()
49 const int kClassDim = logits_shape.dims() - 1; in Compile()
148 const TensorShape logits_shape = ctx->InputShape(0); in Compile() local
150 OP_REQUIRES(ctx, logits_shape.IsSameSize(labels_shape), in Compile()
153 logits_shape.DebugString(), in Compile()
155 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(logits_shape), in Compile()
181 const TensorShape logits_shape = ctx->InputShape(0); in Compile() local
[all …]
Dcategorical_op.cc43 TensorShape logits_shape = ctx->InputShape(0); in Compile() local
46 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(logits_shape), in Compile()
48 logits_shape.DebugString())); in Compile()
54 const int64 dim = logits_shape.dim_size(i); in Compile()
57 errors::InvalidArgument("logits.shape = ", logits_shape.DebugString(), in Compile()
61 const int64 batch_size = logits_shape.dim_size(0); in Compile()
62 const int64 num_classes = logits_shape.dim_size(1); in Compile()
/external/tensorflow/tensorflow/core/ops/
Dstateless_random_ops.cc86 ShapeHandle logits_shape; in __anon530aab340202() local
89 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &logits_shape)); in __anon530aab340202()
92 c->set_output(0, c->Matrix(c->Dim(logits_shape, 0), num_samples)); in __anon530aab340202()
Drandom_ops.cc114 ShapeHandle logits_shape; in __anon979c5d7d0302() local
117 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &logits_shape)); in __anon979c5d7d0302()
120 c->set_output(0, c->Matrix(c->Dim(logits_shape, 0), num_samples)); in __anon979c5d7d0302()
Dboosted_trees_ops.cc206 auto logits_shape = in __anone3ce78ae0702() local
209 c->set_output(0, logits_shape); in __anone3ce78ae0702()
277 auto logits_shape = in __anone3ce78ae0a02() local
280 c->set_output(0, logits_shape); in __anone3ce78ae0a02()
/external/tensorflow/tensorflow/contrib/kernel_methods/python/
Dlosses.py81 logits_shape = array_ops.shape(logits)
82 batch_size, num_classes = logits_shape[0], logits_shape[1]
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad_test.cc116 TensorShape logits_shape({5, 3}); in TEST_F() local
119 auto logits = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(logits_shape)); in TEST_F()
120 auto labels = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(logits_shape)); in TEST_F()
125 RunTest({logits, labels}, {logits_shape, logits_shape}, {y.backprop, y.loss}, in TEST_F()
126 {logits_shape, loss_shape}); in TEST_F()
/external/tensorflow/tensorflow/python/ops/distributions/
Dcategorical.py216 logits_shape = array_ops.shape(self._logits, name="logits_shape")
224 self._event_size = logits_shape[self._batch_rank]
233 self._batch_shape_val = logits_shape[:-1]
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Donehot_categorical.py206 logits_shape = array_ops.shape(math_ops.reduce_sum(logits, -1))
212 ret = array_ops.reshape(ret, logits_shape)
Drelaxed_onehot_categorical.py280 logits_shape = array_ops.shape(math_ops.reduce_sum(logits, axis=[-1]))
294 ret = array_ops.reshape(log_prob, logits_shape)