Searched refs:logits_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 44 TF_ASSIGN_OR_RETURN(xla::Shape logits_shape, b->GetShape(logits)); in BuildSoftmaxCustomCall()
46 logits_shape, {logits_shape}); in BuildSoftmaxCustomCall()
56 const TensorShape logits_shape = ctx->InputShape(0); in Compile() local
57 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(logits_shape), in Compile()
59 logits_shape.DebugString())); in Compile()
63 std::vector<int64> batch_dims(logits_shape.dims() - 1); in Compile()
65 const int kClassDim = logits_shape.dims() - 1; in Compile()
200 const TensorShape logits_shape = ctx->InputShape(0); in Compile() local
202 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(logits_shape), in Compile()
204 logits_shape.DebugString())); in Compile()
[all …]
|
| D | categorical_op.cc | 46 TensorShape logits_shape = ctx->InputShape(0); in Compile() local
49 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(logits_shape), in Compile()
51 logits_shape.DebugString())); in Compile()
57 const int64 dim = logits_shape.dim_size(i); in Compile()
60 errors::InvalidArgument("logits.shape = ", logits_shape.DebugString(), in Compile()
64 const int64 batch_size = logits_shape.dim_size(0); in Compile()
65 const int64 num_classes = logits_shape.dim_size(1); in Compile()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | stateless_random_ops.cc | 105 ShapeHandle logits_shape; in __anona1237d120202() local
108 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &logits_shape)); in __anona1237d120202()
111 c->set_output(0, c->Matrix(c->Dim(logits_shape, 0), num_samples)); in __anona1237d120202()
|
| D | random_ops.cc | 124 ShapeHandle logits_shape; in __anon2645a1db0302() local
127 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &logits_shape)); in __anon2645a1db0302()
130 c->set_output(0, c->Matrix(c->Dim(logits_shape, 0), num_samples)); in __anon2645a1db0302()
|
| D | boosted_trees_ops.cc | 477 auto logits_shape = in __anon770f674c0c02() local
480 c->set_output(0, logits_shape); in __anon770f674c0c02()
555 auto logits_shape = in __anon770f674c0f02() local
558 c->set_output(0, logits_shape); in __anon770f674c0f02()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad_test.cc | 116 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/ |
| D | categorical.py | 216 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/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 2490 TensorShapeProto logits_shape = op_context.op_info.inputs(0).shape(); in PredictSoftmax() local
2491 if (logits_shape.unknown_rank() || logits_shape.dim_size() == 0) { in PredictSoftmax()
2505 EIGEN_COST(scalar_inverse_op<float>) * logits_shape.dim(0).size(); in PredictSoftmax()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 1424 ArrayRef<int64_t> logits_shape = output_type.getShape(); in convertSoftmaxOp() local
1442 RankedTensorType::get(logits_shape, int16_element_qtype); in convertSoftmaxOp()
1444 RankedTensorType::get(logits_shape, rewriter.getIntegerType(32)); in convertSoftmaxOp()
|