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()
|