Home
last modified time | relevance | path

Searched refs:MultiArray (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/executorch/backends/apple/coreml/runtime/delegate/
H A Dmultiarray.mm55 /// Returns `BNNSDataType` from `MultiArray::DataType`.
59 std::optional<BNNSDataType> get_bnns_data_type(MultiArray::DataType datatype) {
61 case MultiArray::DataType::Bool: {
64 case MultiArray::DataType::Byte: {
67 case MultiArray::DataType::Char: {
70 case MultiArray::DataType::Short: {
73 case MultiArray::DataType::Int32: {
76 case MultiArray::DataType::Int64: {
79 case MultiArray::DataType::Float16: {
82 case MultiArray::DataType::Float32: {
[all …]
H A Dcoreml_backend_delegate.mm44 std::optional<MultiArray::DataType> get_data_type(ScalarType scalar_type) {
47 return MultiArray::DataType::Bool;
49 return MultiArray::DataType::Byte;
51 return MultiArray::DataType::Short;
53 return MultiArray::DataType::Int32;
55 return MultiArray::DataType::Int64;
57 return MultiArray::DataType::Float16;
59 return MultiArray::DataType::Float32;
61 return MultiArray::DataType::Float64;
72 std::optional<MultiArray> get_multi_array(EValue *eValue, ArgType argType) {
[all …]
H A Dmultiarray.h35 class MultiArray final {
96 MultiArray(void* data, MemoryLayout layout) : data_(data), layout_(std::move(layout)) { } in MultiArray() function
107 void copy(MultiArray& dst, CopyOptions options = CopyOptions()) const noexcept;
139 std::optional<MLMultiArrayDataType> to_ml_multiarray_data_type(MultiArray::DataType data_type);
142 std::optional<MultiArray::DataType> to_multiarray_data_type(MLMultiArrayDataType data_type);
H A DETCoreMLModelManager.mm120 std::optional<MultiArray::DataType> get_data_type(MLMultiArrayDataType data_type) {
123 return MultiArray::DataType::Float16;
126 return MultiArray::DataType::Float32;
129 return MultiArray::DataType::Float64;
132 return MultiArray::DataType::Int32;
140 void copy(MLMultiArray *src, executorchcoreml::MultiArray& dst) {
146 …MultiArray::MemoryLayout src_layout(get_data_type(src.dataType).value(), to_vector<size_t>(src.sha…
147 MultiArray(const_cast<void *>(bytes), std::move(src_layout)).copy(dst);
151 void set_outputs(std::vector<executorchcoreml::MultiArray>& outputs,
737 argsVec:(const std::vector<executorchcoreml::MultiArray>&)argsVec
[all …]
H A DMLMultiArray_Copy.mm16 MultiArray to_multi_array(void *data,
20 auto layout = MultiArray::MemoryLayout(to_multiarray_data_type(dataType).value(),
23 return MultiArray(data, std::move(layout));
H A DETCoreMLModel.h16 class MultiArray; variable
63 - (nullable NSArray<MLMultiArray*>*)prepareInputs:(const std::vector<executorchcoreml::MultiArray>&…
66 …ray<MLMultiArray*>*)prepareOutputBackings:(const std::vector<executorchcoreml::MultiArray>&)outputs
H A DETCoreMLModel.mm226 - (nullable NSArray<MLMultiArray *> *)prepareArgs:(const std::vector<executorchcoreml::MultiArray>&… argument
262 …MultiArray buffer(mutableBytes, MultiArray::MemoryLayout(to_multiarray_data_type(constraint.dataTy…
275 - (nullable NSArray<MLMultiArray *> *)prepareInputs:(const std::vector<executorchcoreml::MultiArray… argument
285 …y<MLMultiArray *> *)prepareOutputBackings:(const std::vector<executorchcoreml::MultiArray>&)outputs argument
H A DETCoreMLModelManager.h17 class MultiArray; variable
74 argsVec:(const std::vector<executorchcoreml::MultiArray>&)argsVec
H A Dbackend_delegate.h18 class MultiArray; variable
92 const std::vector<MultiArray>& args,
H A Dbackend_delegate.mm107 argsVec:(const std::vector<executorchcoreml::MultiArray>&)argsVec
202 argsVec:(const std::vector<executorchcoreml::MultiArray>&)argsVec
289 const std::vector<MultiArray>& args,
/aosp_15_r20/external/executorch/backends/apple/coreml/runtime/test/
H A DMultiArrayTests.mm25 MultiArray::DataType get_multiarray_data_type();
27 template<> MultiArray::DataType get_multiarray_data_type<float>() {
28 return MultiArray::DataType::Float32;
31 template<> MultiArray::DataType get_multiarray_data_type<double>() {
32 return MultiArray::DataType::Float64;
35 template<> MultiArray::DataType get_multiarray_data_type<int64_t>() {
36 return MultiArray::DataType::Int64;
39 template<> MultiArray::DataType get_multiarray_data_type<int32_t>() {
40 return MultiArray::DataType::Int32;
43 template<> MultiArray::DataType get_multiarray_data_type<int16_t>() {
[all …]
H A DBackendDelegateTests.mm24 MultiArray to_multiarray(MLMultiArray *ml_multiarray) {
27 auto layout = MultiArray::MemoryLayout(to_multiarray_data_type(ml_multiarray.dataType).value(),
35 return MultiArray(bytes, std::move(layout));
38 std::vector<MultiArray> to_multiarrays(NSArray<MLMultiArray *> *ml_multiarrays) {
39 std::vector<MultiArray> result;
/aosp_15_r20/external/executorch/backends/apple/coreml/runtime/sdk/
H A Dmodel_event_logger_impl.mm154 …NSCAssert(scalar_type.has_value(), @"ModelEventLoggerImpl: MultiArray dataType=%ld is not supporte…