Searched refs:boxes_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | crop_and_resize_op_benchmark_test.cc | 30 auto boxes_tensor = boxes.matrix<float>(); in CropAndResize() local
34 boxes_tensor(i, 0) = 0.2; in CropAndResize()
35 boxes_tensor(i, 1) = 0.2; in CropAndResize()
36 boxes_tensor(i, 2) = 0.8; in CropAndResize()
37 boxes_tensor(i, 3) = 0.7; in CropAndResize()
|
| D | mirror_pad_op_benchmark_test.cc | 30 auto boxes_tensor = padding.flat<int>().setZero(); in MirrorPad() local
31 for (int i = 2; i < 6; i++) boxes_tensor(i) = pad; in MirrorPad()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_grad_test_base.py | 526 boxes_tensor = constant_op.constant(boxes, shape=[num_boxes, 4])
529 def crop_resize(image_tensor, boxes_tensor): argument
532 image_tensor, boxes_tensor, box_ind_tensor,
540 lambda x: crop_resize(x, boxes_tensor),
545 [boxes_tensor]))
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5923 nvinfer1::ITensor* boxes_tensor = inputs.at(0).tensor(); in ConvertCombinedNMS() local
5931 const auto boxes_dims = boxes_tensor->getDimensions(); in ConvertCombinedNMS()
6055 plugin_inputs.push_back(boxes_tensor); in ConvertCombinedNMS()
|
| D | convert_nodes_test.cc | 3135 auto boxes_tensor = ops::Placeholder(s.WithOpName("boxes"), tf_type); in TEST_P() local
3150 s.WithOpName("my_nms"), boxes_tensor, scores_tensor, in TEST_P()
|