| /aosp_15_r20/external/ComputeLibrary/src/core/ |
| H A D | TensorInfo.cpp | 93 TensorInfo::TensorInfo(const TensorShape &tensor_shape, Format format) in TensorInfo()
105 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type) in TensorInfo()
111 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, Qu… in TensorInfo()
118 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, Da… in TensorInfo()
130 void TensorInfo::init(const TensorShape &tensor_shape, Format format) in init()
140 void TensorInfo::init(const TensorShape &tensor_shape, Format format, in init()
157 void TensorInfo::init(const TensorShape &tensor_shape, size_t num_channels, DataType data_type) in init()
168 void TensorInfo::init(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, in init()
185 size_t TensorInfo::init_auto_padding(const TensorShape &tensor_shape, Format format) in init_auto_padding()
196 size_t TensorInfo::init_auto_padding(const TensorShape &tensor_shape, size_t num_channels, DataType… in init_auto_padding()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | shape_util.cc | 89 TensorShape* tensor_shape) { in XLAShapeToTensorShape()
104 const PartialTensorShape& tensor_shape, in TensorShapeToXLAShape()
113 const PartialTensorShape& tensor_shape) { in TensorShapeToXLAShape()
137 Status TensorShapeToXLAShape(DataType dtype, const TensorShape& tensor_shape, in TensorShapeToXLAShape()
146 const TensorShape& tensor_shape) { in TensorShapeToXLAShape()
153 const TensorShape& tensor_shape) { in TensorShapeToXLAShape()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_rnn_1_3.mod.py | 20 def convert_to_time_major(tensor, tensor_shape): argument
30 def reverse_batch_major(tensor, tensor_shape): argument
33 def split_tensor_in_two(tensor, tensor_shape): argument
|
| D | bidirectional_sequence_rnn_state_output.mod.py | 20 def convert_to_time_major(tensor, tensor_shape): argument
31 def reverse_batch_major(tensor, tensor_shape): argument
35 def split_tensor_in_two(tensor, tensor_shape): argument
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | bidirectional_sequence_rnn.mod.py | 20 def convert_to_time_major(tensor, tensor_shape): argument
30 def reverse_batch_major(tensor, tensor_shape): argument
33 def split_tensor_in_two(tensor, tensor_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | quantization_util.cc | 35 const RuntimeShape &tensor_shape, int32_t zero_point, in DequantizeInt8()
46 const RuntimeShape& tensor_shape, in PerChannelDequantizeInt8()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| H A D | ragged_tensor_to_tensor_tflite.cc | 118 const RuntimeShape tensor_shape(tensor.dims->size, tensor.dims->data); in TensorShapeFromTensor() local
156 const RuntimeShape tensor_shape(tensor->dims->size, tensor->dims->data); in GetMaxWidthValueRowID() local
187 const RuntimeShape tensor_shape(tensor->dims->size, tensor->dims->data); in GetMaxWidthRowSplit() local
335 const RuntimeShape tensor_shape(value_rowids.dims->size, in CalculateOutputIndexValueRowID() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | list_ops.cc | 185 shape_inference::ShapeHandle tensor_shape = c->UnknownShape(); in __anon5e0c1ea50502() local
344 shape_inference::ShapeHandle tensor_shape = c->input(0); in __anon5e0c1ea50902() local
382 shape_inference::ShapeHandle tensor_shape = c->input(0); in __anon5e0c1ea50a02() local
|
| H A D | data_flow_ops.cc | 756 ShapeHandle tensor_shape = shapes->at(0).shape; in __anon1926cf7a1902() local
781 ShapeHandle tensor_shape = shapes->at(0).shape; in __anon1926cf7a1a02() local
821 ShapeHandle tensor_shape = shapes->at(0).shape; in __anon1926cf7a1b02() local
|
| /aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama2/runner/ |
| H A D | runner.cpp | 295 auto tensor_shape = tensor_meta->sizes(); in generate() local
324 auto tensor_shape = tensor_meta->sizes(); in generate() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | tensor.cpp | 80 auto tensor_shape = shape(); in size() local
158 auto tensor_shape = shape(); in SetInPlaceIrValue() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | tensor_format.h | 468 inline int64_t GetTensorDim(const TensorShape& tensor_shape, in GetTensorDim()
476 inline int64_t GetFilterDim(const TensorShape& tensor_shape, in GetFilterDim()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | costmodel.cc | 254 const TensorShapeProto& tensor_shape, in RecordMaxMemorySize()
558 Bytes CostModel::MinTensorMemoryUsage(const TensorShapeProto& tensor_shape, in MinTensorMemoryUsage()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | fake_param_op.cc | 34 PartialTensorShape tensor_shape; in XlaFakeParamOp() local
|
| H A D | sendrecv_ops.cc | 68 TensorShape tensor_shape; in RecvOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset.proto | 18 .tensorflow.TensorShapeProto tensor_shape = 2; field
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/ |
| H A D | SubTensor.cpp | 35 SubTensor::SubTensor(ITensor *parent, const TensorShape &tensor_shape, const Coordinates &coords, b… in SubTensor()
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/mutators/ |
| H A D | NodeFusionMutator.cpp | 564 const TensorShape tensor_shape = input_edge->tensor()->desc().shape; in fuse_convolution_with_post_ops() local
662 const TensorShape tensor_shape = input_edge->tensor()->desc().shape; in fuse_convolution_batch_normalization_with_post_ops() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/protobuf/ |
| H A D | projector_config.proto | 36 repeated uint32 tensor_shape = 4; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
| H A D | snapshot.proto | 39 .tensorflow.TensorShapeProto tensor_shape = 2; field
|
| /aosp_15_r20/external/ComputeLibrary/tests/datasets/ |
| H A D | ROIDataset.h | 93 void add_config(TensorShape tensor_shape, ROIPoolingLayerInfo info, TensorShape rois_shape) in add_config()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/ |
| H A D | CLSubTensor.cpp | 37 CLSubTensor::CLSubTensor(ICLTensor *parent, const TensorShape &tensor_shape, const Coordinates &coo… in CLSubTensor()
|
| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A D | graph_lenet.cpp | 70 …const TensorShape tensor_shape = permute_shape(TensorShape(28U, 28U, 1U, batches), DataLayout:… in do_setup() local
|
| H A D | graph_vgg19.cpp | 69 …const TensorShape tensor_shape = permute_shape(TensorShape(224U, 224U, 3U, common_params.batch… in do_setup() local
|
| H A D | graph_alexnet.cpp | 77 …const TensorShape tensor_shape = permute_shape(TensorShape(227U, 227U, 3U, common_params.batch… in do_setup() local
|