Searched defs:image_shape (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | BoundingBoxTransformFixture.h | 45 std::vector<float> generate_deltas(std::vector<float> &boxes, const TensorShape &image_shape, size_… in generate_deltas()
79 std::vector<float> generate_boxes(const TensorShape &image_shape, size_t num_boxes, std::mt19937 &g… in generate_boxes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | morphological_ops_test.py | 183 def _ConstructAndTestGradient(self, image_shape, kernel_shape, strides, rates, argument
489 def _ConstructAndTestGradient(self, image_shape, kernel_shape, strides, rates, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | extract_jpeg_shape_op.cc | 57 Tensor* image_shape = nullptr; in Compute() local
|
| /aosp_15_r20/external/executorch/examples/models/llava/ |
| H A D | main.cpp | 94 std::array<int32_t, 3> image_shape = {3, 240, 336}; in main() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| H A D | resnet50_graph_test.py | 29 def image_shape(batch_size): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | image_grad.cc | 119 auto image_shape = Shape(scope, op.input(0)); in CropAndResizeGradHelper() local
|
| /aosp_15_r20/external/executorch/examples/qualcomm/ |
| H A D | utils.py | 375 dataset_path, data_size, image_shape, crop_size=None, shuffle=True argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | fbgemm_utils.cpp | 106 const std::vector<int>& image_shape, in MakeFbgemmConvParam()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator.cc | 950 auto image_shape = MaybeGetMinimumShape(original_image_shape, in ConvolutionDimensionsFromInputs() local
2162 auto image_shape = in OpDimensionsFromInputs() local
|
| /aosp_15_r20/external/pytorch/torch/backends/_nnapi/ |
| H A D | serializer.py | 268 def get_conv_pool_shape(image_shape, args, out_ch, transpose): argument
|