| /aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
| H A D | LayerSupportBase.cpp | 85 bool LayerSupportBase::IsActivationSupported(const TensorInfo&, // input in IsActivationSupported()
93 bool LayerSupportBase::IsAdditionSupported(const TensorInfo&, // input0 in IsAdditionSupported()
101 bool LayerSupportBase::IsArgMinMaxSupported(const armnn::TensorInfo&, // input in IsArgMinMaxSupported()
109 bool LayerSupportBase::IsBatchNormalizationSupported(const TensorInfo&, //input in IsBatchNormalizationSupported()
121 bool LayerSupportBase::IsBatchToSpaceNdSupported(const TensorInfo&, // input in IsBatchToSpaceNdSupported()
129 bool LayerSupportBase::IsCastSupported(const TensorInfo&, //input in IsCastSupported()
136 bool LayerSupportBase::IsChannelShuffleSupported(const TensorInfo&, //input in IsChannelShuffleSupported()
144 bool LayerSupportBase::IsComparisonSupported(const TensorInfo&, // input0 in IsComparisonSupported()
153 bool LayerSupportBase::IsConcatSupported(const std::vector<const TensorInfo*>, // inputs in IsConcatSupported()
154 const TensorInfo&, // output in IsConcatSupported()
[all …]
|
| H A D | WorkloadFactoryBase.hpp | 28 std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& /*tensorInfo*/, in CreateTensorHandle() argument
32 std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& /*tensorInfo*/, in CreateTensorHandle() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/model_modifier/ |
| H A D | validation_graph_builder.h | 130 struct TensorInfo { struct
131 ~TensorInfo() { std::free(jpeg_buffer_contents); } in ~TensorInfo() argument
133 std::vector<int32_t> entrypoint_inputs;
134 std::vector<int32_t> entrypoint_outputs;
135 std::vector<int32_t> jpeg_images;
140 std::vector<int32_t> quantized_images;
141 std::vector<int32_t> float_images;
143 std::vector<int32_t> main_outputs; // First half of validation_inputs.
144 std::vector<int32_t> validation_inputs;
150 std::vector<int32_t> dequantized_validation_inputs;
[all …]
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/codegen/ |
| H A D | android_java_generator.h | 36 struct TensorInfo { struct
58 std::vector<TensorInfo> inputs; argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| H A D | embedder.cc | 136 struct TensorInfo { struct in tflite::acceleration::__anonb0e9eaf00111::ValidationGraphBuilder
137 std::vector<int32_t> entrypoint_inputs;
138 std::vector<int32_t> entrypoint_outputs;
139 std::vector<int32_t> jpeg_images;
144 std::vector<int32_t> quantized_images;
145 std::vector<int32_t> float_images;
147 std::vector<int32_t> main_outputs; // First half of validation_inputs.
148 std::vector<int32_t> validation_inputs;
154 std::vector<int32_t> dequantized_validation_inputs;
155 std::vector<int32_t> validation_outputs;
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/ |
| H A D | TensorInfo.cuh | 11 struct TensorInfo { struct
28 __host__ __device__ inline bool isContiguous() const { in isContiguous()
32 T* data;
33 IndexType sizes[MAX_TENSORINFO_DIMS];
34 IndexType strides[MAX_TENSORINFO_DIMS];
35 int dims;
39 TensorInfo<T, IndexType>::TensorInfo() { in TensorInfo() function in at::cuda::detail::TensorInfo
45 TensorInfo<T, IndexType>::TensorInfo(T* p, in TensorInfo() function in at::cuda::detail::TensorInfo
|
| /aosp_15_r20/external/ComputeLibrary/src/core/ |
| H A D | TensorInfo.cpp | 36 TensorInfo::TensorInfo() in TensorInfo() function in arm_compute::TensorInfo
42 TensorInfo::TensorInfo(const ITensorInfo &info) in TensorInfo() function in arm_compute::TensorInfo
63 TensorInfo::TensorInfo(const TensorInfo &info) in TensorInfo() function in arm_compute::TensorInfo
83 TensorInfo::TensorInfo(Format format) in TensorInfo() function in arm_compute::TensorInfo
88 TensorInfo::TensorInfo(unsigned int width, unsigned int height, Format format) in TensorInfo() function in arm_compute::TensorInfo
93 TensorInfo::TensorInfo(const TensorShape &tensor_shape, Format format) in TensorInfo() function in arm_compute::TensorInfo
99 TensorInfo::TensorInfo(size_t num_channels, DataType data_type) in TensorInfo() function in arm_compute::TensorInfo
105 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type) in TensorInfo() function in arm_compute::TensorInfo
111 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, Qu… in TensorInfo() function in arm_compute::TensorInfo
118 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, Da… in TensorInfo() function in arm_compute::TensorInfo
|
| /aosp_15_r20/external/armnn/include/armnnTestUtils/ |
| H A D | MockBackend.hpp | 333 bool IsInputSupported(const TensorInfo& /*input*/, in IsInputSupported() argument
339 bool IsOutputSupported(const TensorInfo& /*input*/, in IsOutputSupported() argument
345 bool IsAdditionSupported(const TensorInfo& /*input0*/, in IsAdditionSupported() argument
353 bool IsConvolution2dSupported(const TensorInfo& /*input*/, in IsConvolution2dSupported() argument
357 const Optional<TensorInfo>& /*biases*/, in IsConvolution2dSupported()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/ |
| H A D | tensor_info.h | 11 struct TORCH_API TensorInfo { struct
12 uint32_t* sizes(size_t nDim) { in sizes()
15 uint32_t* strides(size_t nDim) { in strides()
19 void* data;
21 uint32_t sizes_strides[0];
|
| /aosp_15_r20/external/tensorflow/tensorflow/go/ |
| H A D | signature.go | 87 type TensorInfo struct { struct
88 Name string
89 DType DataType
90 Shape Shape
109 func tensorInfoFromProto(pb *corepb.TensorInfo) TensorInfo {
|
| H A D | signature_test.go | 171 func diffTensorInfos(a, b TensorInfo) (string, error) {
|
| /aosp_15_r20/external/armnn/src/armnn/ |
| H A D | Tensor.cpp | 341 TensorInfo::TensorInfo() in TensorInfo() function in armnn::TensorInfo
346 TensorInfo::TensorInfo(const TensorShape& shape, in TensorInfo() function in armnn::TensorInfo
359 TensorInfo::TensorInfo(unsigned int numDimensions, in TensorInfo() function in armnn::TensorInfo
371 TensorInfo::TensorInfo(const TensorShape& shape, in TensorInfo() function in armnn::TensorInfo
384 TensorInfo::TensorInfo(unsigned int numDimensions, in TensorInfo() function in armnn::TensorInfo
398 TensorInfo::TensorInfo(const TensorInfo& other) in TensorInfo() function in armnn::TensorInfo
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/ |
| H A D | Interpreter.cs | 44 public struct TensorInfo { struct in TensorFlowLite.Interpreter
45 public string name { get; internal set; }
46 public DataType type { get; internal set; }
47 public int[] dimensions { get; internal set; }
48 public QuantizationParams quantizationParams { get; internal set; }
50 public override string ToString() { in ToString()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | decode_proto_op.cc | 883 struct TensorInfo { in AccumulateFields() struct
884 explicit TensorInfo(Tensor* tensor) { in AccumulateFields() function
906 DataType dtype; in AccumulateFields()
907 int last_dim_size; in AccumulateFields()
908 int stride; in AccumulateFields()
909 uint8* data; in AccumulateFields()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/ |
| H A D | misc.h | 9 struct TensorInfo { struct
13 std::optional<TensorInfo> getTensorInfo(const BufHandle& b); argument
|
| /aosp_15_r20/external/armnn/src/armnn/test/ |
| H A D | OptimizerTests.cpp | 168 bool IsInputSupported(const TensorInfo& /*input*/, in IsInputSupported() argument
174 bool IsOutputSupported(const TensorInfo& /*input*/, in IsOutputSupported() argument
180 bool IsActivationSupported(const TensorInfo& /*input0*/, in IsActivationSupported() argument
|
| /aosp_15_r20/external/pytorch/functorch/csrc/dim/ |
| H A D | dim.cpp | 784 struct TensorInfo { struct
785 TensorRef tensor;
786 Slice<DimEntry> levels;
787 bool has_device;
788 TensorRef batchedtensor;
789 int64_t ndim() const { in ndim()
792 operator bool() const { in operator bool()
796 …ic TensorInfo create(Arena& A, mpy::handle h, bool ensure_batched=true, bool ensure_present=true) { in create()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/ |
| H A D | TensorAllocator.h | 39 class TensorInfo; variable
|
| H A D | OperatorTensor.h | 36 class TensorInfo; variable
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/UNIT/ |
| H A D | TensorInfo.cpp | 39 TEST_SUITE(TensorInfo) in TEST_SUITE() argument
|
| /aosp_15_r20/external/executorch/runtime/executor/ |
| H A D | method_meta.cpp | 68 TensorInfo::TensorInfo( in TensorInfo() function in executorch::runtime::TensorInfo
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
| H A D | meta_graph.proto | 214 message TensorInfo { message
|
| /aosp_15_r20/external/armnn/include/armnn/ |
| H A D | Tensor.hpp | 152 class TensorInfo class
|
| /aosp_15_r20/external/armnn/src/backends/reference/ |
| H A D | RefLayerSupport.cpp | 1622 bool RefLayerSupport::IsInputSupported(const TensorInfo& /*input*/, in IsInputSupported() argument
2089 bool RefLayerSupport::IsOutputSupported(const TensorInfo& /*output*/, in IsOutputSupported() argument
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
| D | meta_graph.pb.cc | 203 PROTOBUF_CONSTEXPR TensorInfo::TensorInfo( in TensorInfo() function in tensorflow::TensorInfo
3371 TensorInfo::TensorInfo(::PROTOBUF_NAMESPACE_ID::Arena* arena, in TensorInfo() function in tensorflow::TensorInfo
3377 TensorInfo::TensorInfo(const TensorInfo& from) in TensorInfo() function in tensorflow::TensorInfo
|