Searched refs:boxes_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | combined_nms_test.py | 71 boxes_shape = [batch_size, num_boxes, q, 4]
78 [boxes_shape, scores_shape], [
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 409 const TensorShape& boxes_shape = context->InputShape("boxes"); in Compile() local
410 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(boxes_shape), in Compile()
412 boxes_shape.DebugString())); in Compile()
413 const int64 num_boxes = boxes_shape.dim_size(0); in Compile()
414 OP_REQUIRES(context, boxes_shape.dim_size(1) == 4, in Compile()
416 boxes_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cc | 535 TensorShape boxes_shape({num_batches, per_batch_size, 4}); in BatchedNonMaxSuppressionOp() local
537 context->allocate_output(0, boxes_shape, &nmsed_boxes_t)); in BatchedNonMaxSuppressionOp()
|