Searched refs:scores_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | dense_attention.py | 165 scores_shape = array_ops.shape(scores)
168 [array_ops.ones_like(scores_shape[:-2]), scores_shape[-2:]],
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | combined_nms_test.py | 72 scores_shape = [batch_size, num_boxes, num_classes]
78 [boxes_shape, scores_shape], [
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 417 const TensorShape& scores_shape = context->InputShape("scores"); in Compile() local
418 OP_REQUIRES(context, TensorShapeUtils::IsVector(scores_shape), in Compile()
420 scores_shape.DebugString())); in Compile()
422 context, scores_shape.dim_size(0) == num_boxes, in Compile()
424 scores_shape.DebugString())); in Compile()
569 XlaGather(indices_sorted, scores_shape, selected_indices_sorted, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cc | 541 TensorShape scores_shape({num_batches, per_batch_size}); in BatchedNonMaxSuppressionOp() local
543 context->allocate_output(1, scores_shape, &nmsed_scores_t)); in BatchedNonMaxSuppressionOp()
548 context->allocate_output(2, scores_shape, &nmsed_classes_t)); in BatchedNonMaxSuppressionOp()
|