Searched refs:labels_shape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 149 const TensorShape labels_shape = ctx->InputShape(1); in Compile() local
150 OP_REQUIRES(ctx, logits_shape.IsSameSize(labels_shape), in Compile()
154 " labels_size=", labels_shape.DebugString())); in Compile()
182 const TensorShape labels_shape = ctx->InputShape(1); in Compile() local
186 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(labels_shape), in Compile()
188 labels_shape.DebugString())); in Compile()
189 OP_REQUIRES(ctx, logits_shape.dim_size(0) == labels_shape.dim_size(0), in Compile()
194 labels_shape.DebugString())); in Compile()
213 builder, depth, /*axis=*/1, input_type(1), labels_shape, in Compile()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | confusion_matrix.py | 65 labels_shape = labels.get_shape()
66 labels_rank = labels_shape.ndims
85 labels_shape.dims[-1].is_compatible_with(1)):
|
| D | metrics_impl.py | 2967 labels_shape = array_ops.shape(labels)
2968 labels_size = labels_shape[-1]
2970 return array_ops.fill(labels_shape[0:-1], num_relevant_scalar, name=scope)
|
| D | nn_ops.py | 3314 labels_shape = array_ops.shape(labels)
3359 cost = array_ops.reshape(cost, labels_shape)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | rnn_common.py | 233 labels_shape = array_ops.shape(labels)
234 batch_size = labels_shape[0]
235 padded_length = labels_shape[1]
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | tensor_util.py | 105 labels_shape = labels.get_shape()
106 labels_rank = labels_shape.ndims
125 labels_shape.dims[-1].is_compatible_with(1)):
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_loss_op.cc | 101 TensorShape labels_shape({batch_size, max_label_len}); in Compute() local
106 labels_shape, order, &labels_sp)); in Compute()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/utils/ |
| D | losses.py | 118 labels_shape = labels.get_shape()
119 if len(labels_shape) != 2:
|