Searched refs:scores_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 415 const TensorShape& scores_shape = context->InputShape("scores"); in Compile() local
416 OP_REQUIRES(context, TensorShapeUtils::IsVector(scores_shape), in Compile()
418 scores_shape.DebugString())); in Compile()
420 context, scores_shape.dim_size(0) == num_boxes, in Compile()
422 scores_shape.DebugString())); in Compile()
563 XlaGather(indices_sorted, scores_shape, selected_indices_sorted, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | non_max_suppression_op.cc | 407 TensorShape scores_shape({num_batches, per_batch_size}); in BatchedNonMaxSuppressionOp() local
409 context->allocate_output(1, scores_shape, &nmsed_scores_t)); in BatchedNonMaxSuppressionOp()
414 context->allocate_output(2, scores_shape, &nmsed_classes_t)); in BatchedNonMaxSuppressionOp()
|