| /aosp_15_r20/external/armnn/src/armnn/ |
| H A D | Tensor.cpp | 25 TensorShape::TensorShape() in TensorShape() function in armnn::TensorShape
30 TensorShape::TensorShape(unsigned int numDimensions, bool initDimensionsSpecificity) in TensorShape() function in armnn::TensorShape
40 TensorShape::TensorShape(const unsigned int numDimensions, const unsigned int* const dimensionSizes) in TensorShape() function in armnn::TensorShape
54 TensorShape::TensorShape(std::initializer_list<unsigned int> dimensionSizeList) in TensorShape() function in armnn::TensorShape
59 TensorShape::TensorShape(unsigned int numDimensions, in TensorShape() function in armnn::TensorShape
80 TensorShape::TensorShape(std::initializer_list<unsigned int> dimensionSizeList, in TensorShape() function in armnn::TensorShape
92 TensorShape::TensorShape(Dimensionality dimensionality) in TensorShape() function in armnn::TensorShape
116 TensorShape::TensorShape(const TensorShape& other) in TensorShape() function in armnn::TensorShape
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_embedding_v2.py | 424 per_replica_input_shapes: Optional[List[TensorShape]] = None,
477 self, input_shapes: List[TensorShape]) -> List[TensorShape]:
1408 def _update_output_shapes(self, incoming_output_shapes: List[TensorShape]):
1427 def _check_output_shapes(self, incoming_output_shapes: List[TensorShape]):
1464 def _is_tensor_shape_match(self, shape_a: TensorShape,
1465 shape_b: TensorShape) -> bool:
|
| H A D | tpu_embedding_v2_utils.py | 1012 output_shape: Optional[Union[List[int], TensorShape]] = None, argument
|
| H A D | tpu.py | 1034 padded_shapes: List[Optional[tensor_shape.TensorShape]],
|
| /aosp_15_r20/external/federated-compute/fcp/aggregation/core/ |
| H A D | tensor_shape.h | 42 TensorShape(ForwardIterator first, ForwardIterator last) in TensorShape() function
45 TensorShape(std::initializer_list<size_t> dim_sizes) in TensorShape() function
74 explicit TensorShape(DimSizesVector&& dim_sizes) in TensorShape() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | dynamic_ragged_shape.py | 1118 static_inner_shape: tensor_shape.TensorShape,
3191 static_inner_shape: tensor_shape.TensorShape,
3253 old_shape: tensor_shape.TensorShape,
3271 def _batch_tensor_shape(old_shape: tensor_shape.TensorShape,
3277 old_shape: tensor_shape.TensorShape,
|
| H A D | dynamic_ragged_shape_test.py | 2277 def assertDimsEqual(self, x: tensor_shape.TensorShape,
2278 y: tensor_shape.TensorShape):
3101 def assertTensorShapeEqual(self, a: tensor_shape.TensorShape,
3102 b: tensor_shape.TensorShape) -> None:
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/ |
| H A D | CLUtils.h | 33 class TensorShape; variable
|
| /aosp_15_r20/external/federated-compute/fcp/artifact_building/ |
| H A D | proto_helpers.py | 27 t: tf.Tensor, shape_hint: Optional[tf.TensorShape] = None
|
| H A D | tensor_utils.py | 118 tensor_shape: Optional[tf.TensorShape] = None,
|
| /aosp_15_r20/external/executorch/examples/mediatek/executor_runner/llama_runner/ |
| H A D | LlamaModelChunk.h | 35 using TensorShape = executorch::runtime::Span<const int32_t>; variable
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/ |
| H A D | TensorShape.h | 47 TensorShape(Ts... dims) in TensorShape() function
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Analysis/ |
| H A D | TensorSpec.cpp | 86 std::vector<int64_t> TensorShape; in getTensorSpecFromJSON() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | tfrt_op_kernel.h | 57 class TensorShape; variable
|
| /aosp_15_r20/external/armnn/include/armnn/ |
| H A D | Tensor.hpp | 20 class TensorShape class
|
| /aosp_15_r20/external/ComputeLibrary/utils/ |
| H A D | TypePrinter.h | 3371 inline std::string to_string(const std::tuple<TensorShape, TensorShape, arm_compute::WeightFormat> … in to_string()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor_shape.h | 36 class TensorShape; variable
|
| H A D | tensor_shape.cc | 843 MAKE_SHAPE(int32, TensorShape) in MAKE_SHAPE() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | tensor_shape.py | 742 class TensorShape(trace.TraceType, trace_type.Serializable): class
|