Searched refs:boxes_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 407 const TensorShape& boxes_shape = context->InputShape("boxes"); in Compile() local
408 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(boxes_shape), in Compile()
410 boxes_shape.DebugString())); in Compile()
411 const int64 num_boxes = boxes_shape.dim_size(0); in Compile()
412 OP_REQUIRES(context, boxes_shape.dim_size(1) == 4, in Compile()
414 boxes_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_grad_test.py | 420 boxes_shape = [num_boxes, 4]
450 [image_tensor, boxes_tensor], [image_shape, boxes_shape],
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | non_max_suppression_op.cc | 401 TensorShape boxes_shape({num_batches, per_batch_size, 4}); in BatchedNonMaxSuppressionOp() local
403 context->allocate_output(0, boxes_shape, &nmsed_boxes_t)); in BatchedNonMaxSuppressionOp()
|