Searched refs:num_input_tensors (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_foreach.py | 853 num_input_tensors=[2],
888 num_input_tensors=[2],
899 num_input_tensors=[2],
936 num_input_tensors=[3],
946 num_input_tensors=[3],
987 num_input_tensors=[N],
1096 dtype=dtype, device=device, num_input_tensors=[2], same_size=True
1135 num_input_tensors=[2],
1186 device, dtype, requires_grad=True, num_input_tensors=[1]
1379 num_input_tensors=[5],
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/ |
| H A D | serialization_test.cc | 65 int num_input_tensors, in GenerateTfLiteDelegateParams() argument
69 auto* input_tensors = TfLiteIntArrayCreate(num_input_tensors); in GenerateTfLiteDelegateParams()
77 for (int i = 0; i < num_input_tensors; ++i) { in GenerateTfLiteDelegateParams()
|
| /aosp_15_r20/hardware/google/pixel/thermal/virtualtemp_estimator/ |
| D | virtualtemp_estimator_data.h | 33 typedef void *(*tflitewrapper_create)(int num_input_tensors, int num_output_tensors);
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/ |
| H A D | batch_resource_base.cc | 551 const int num_input_tensors = input_task.inputs.size(); in SplitInputTask() local
555 for (int i = 0; i < num_input_tensors; ++i) { in SplitInputTask()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 9507 …num_input_tensors = kwargs.pop("num_input_tensors") if num_input_tensors_specified else foreach_nu…
9508 assert isinstance(num_input_tensors, list)
9516 num_input_tensors, self._rightmost_arg_types, self._intersperse_empty):
9517 … if intersperse_empty_tensors and (num_tensors != max(num_input_tensors) or str(device) == 'cpu'):
9581 num_input_tensors = kwargs.pop("num_input_tensors", foreach_num_tensors)
9582 assert isinstance(num_input_tensors, list)
9588 num_input_tensors,
9646 …num_input_tensors = kwargs.pop("num_input_tensors") if num_input_tensors_specified else foreach_nu…
9647 assert isinstance(num_input_tensors, list)
9652 …for num_tensors, rightmost_arg_type in itertools.product(num_input_tensors, self._rightmost_arg_ty…
|