Home
last modified time | relevance | path

Searched defs:boxes_shape (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DCropResizeFixture.h50 …void setup(TensorShape src_shape, TensorShape boxes_shape, Coordinates2D crop_size, InterpolationP… in setup()
70 …TensorType compute_target(const TensorShape &src_shape, const TensorShape &boxes_shape, const Coor… in compute_target()
108 …ensor<float> compute_reference(const TensorShape &src_shape, const TensorShape &boxes_shape, const… in compute_reference()
/aosp_15_r20/external/ComputeLibrary/tests/datasets/
H A DCropResizeDataset.h101 …void add_config(TensorShape src_shape, TensorShape boxes_shape, Coordinates2D crop_size, Interpola… in add_config()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dimage_ops.cc425 const TensorShape& boxes_shape = context->InputShape("boxes"); in ComputeResult() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dnon_max_suppression_op.cc558 TensorShape boxes_shape({num_batches, per_batch_size, 4}); in BatchedNonMaxSuppressionOp() local