| /aosp_15_r20/external/executorch/examples/models/llava/ |
| H A D | main.cpp | 105 torch::Tensor image_tensor; in main() local
106 torch::load(image_tensor, image_path); // CHW in main()
110 image_tensor.size(0), in main()
111 image_tensor.size(1), in main()
112 image_tensor.size(2)); in main()
114 image_tensor.data_ptr<uint8_t>(), in main()
115 image_tensor.data_ptr<uint8_t>() + image_tensor.numel()); in main()
118 .width = static_cast<int32_t>(image_tensor.size(2)), in main()
119 .height = static_cast<int32_t>(image_tensor.size(1))}}; in main()
|
| H A D | image_util.py | 73 image_tensor = prepare_image(image, target_h=336, target_w=336)
74 serialize_image(image_tensor, args.output_path)
|
| /aosp_15_r20/external/executorch/examples/models/llama3_2_vision/preprocess/ |
| H A D | test_preprocess.py | 152 image_tensor = F.to_dtype(
158 image_tensor = image_tensor + 0
161 image_tensor = image_tensor.contiguous()
176 image=image_tensor,
185 image_tensor.shape[-2:], best_resolution, max_size
189 return image_tensor, inscribed_size, best_resolution
246 image_tensor, inscribed_size, best_resolution = self.prepare_inputs(
253 image_tensor, inscribed_size, best_resolution
264 image_tensor, inscribed_size, best_resolution
276 (image_tensor, inscribed_size, best_resolution)
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | morphological_ops_test.py | 204 image_tensor = constant_op.constant(image, shape=image_shape, name="input")
208 def compute_dilation2d(image_tensor, kernel_tensor): argument
210 image_tensor,
222 lambda x: compute_dilation2d(x, kernel_tensor), [image_tensor]))
225 lambda x: compute_dilation2d(image_tensor, x), [kernel_tensor]))
510 image_tensor = constant_op.constant(image, shape=image_shape, name="input")
514 def compute_erosion2d(image_tensor, kernel_tensor): argument
516 image_tensor,
528 lambda x: compute_erosion2d(x, kernel_tensor), [image_tensor]))
531 lambda x: compute_erosion2d(image_tensor, x), [kernel_tensor]))
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | image_grad_test_base.py | 523 image_tensor = constant_op.constant(image, shape=image_shape)
527 def crop_resize(image_tensor, boxes_tensor): argument
530 image_tensor, boxes_tensor, box_ind_tensor,
541 [image_tensor])
544 lambda x: crop_resize(image_tensor, x),
550 [image_tensor]))
553 lambda x: crop_resize(image_tensor, x),
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/ |
| H A D | convert_image_to_csv.py | 61 image_tensor = image_ops.decode_image(file_data, channels=channels).eval()
62 resized_tensor = image_ops.resize_images_v2(image_tensor,
|
| /aosp_15_r20/external/executorch/examples/models/llava/runner/ |
| H A D | llava_image_prefiller.h | 32 auto image_tensor = executorch::extension::from_blob( in prefill() local
38 ET_UNWRAP(module_->execute(kImageEncoderMethod, image_tensor)); in prefill()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_resize_bilinear_op_test.cc | 191 void TestResizeBilinear(const Tensor& image_tensor, const DataType dt, in TestResizeBilinear() argument
217 TF_EXPECT_OK(session.Run({{placeholder, image_tensor}}, in TestResizeBilinear()
232 << TensorShape(image_tensor.shape()).DebugString() in TestResizeBilinear()
|