Home
last modified time | relevance | path

Searched defs:GetShape (Results 1 – 25 of 32) sorted by relevance

12

/aosp_15_r20/external/armnn/include/armnn/
H A DTensor.hpp191 const TensorShape& GetShape() const { return m_Shape; } in GetShape() function in armnn::TensorInfo
192 TensorShape& GetShape() { return m_Shape; } in GetShape() function in armnn::TensorInfo
297 const TensorShape& GetShape() const { return m_Info.GetShape(); } in GetShape() function in armnn::BaseTensor
298 TensorShape& GetShape() { return m_Info.GetShape(); } in GetShape() function in armnn::BaseTensor
/aosp_15_r20/external/armnn/src/backends/cl/
H A DClTensorHandle.hpp102 TensorShape GetShape() const override in GetShape() function in armnn::ClTensorHandle
270 TensorShape GetShape() const override in GetShape() function in armnn::ClSubTensorHandle
H A DClImportTensorHandle.hpp77 TensorShape GetShape() const override in GetShape() function in armnn::ClImportTensorHandle
379 TensorShape GetShape() const override in GetShape() function in armnn::ClImportSubTensorHandle
/aosp_15_r20/external/armnn/src/backends/neon/
H A DNeonTensorHandle.hpp97 TensorShape GetShape() const override in GetShape() function in armnn::NeonTensorHandle
317 TensorShape GetShape() const override in GetShape() function in armnn::NeonSubTensorHandle
/aosp_15_r20/external/armnn/src/backends/reference/
H A DRefTensorHandle.hpp44 TensorShape GetShape() const override in GetShape() function in armnn::RefTensorHandle
/aosp_15_r20/external/armnn/src/dynamic/sample/
H A DSampleTensorHandle.hpp44 armnn::TensorShape GetShape() const override in GetShape() function in sdb::SampleTensorHandle
/aosp_15_r20/external/armnn/src/backends/tosaReference/
H A DTosaRefTensorHandle.hpp44 TensorShape GetShape() const override in GetShape() function in armnn::TosaRefTensorHandle
/aosp_15_r20/external/armnn/include/armnnTestUtils/
H A DMockTensorHandle.hpp49 TensorShape GetShape() const override in GetShape() function in armnn::MockTensorHandle
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_expression_test.cc93 TEST_F(XlaExpressionTest, GetShape) { in TEST_F() argument
H A Dxla_expression.cc211 StatusOr<TensorShape> XlaExpression::GetShape() const { in GetShape() function in tensorflow::XlaExpression
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Delementwise_spmd_expander.cc50 StatusOr<llvm::SmallVector<int64_t, 4>> GetShape(mlir::Value value) { in GetShape() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/rpc/
H A Dxla_service.proto74 rpc GetShape(GetShapeRequest) returns (GetShapeResponse) { rpc
H A Dgrpc_service.cc127 ::grpc::Status GRPCService::GetShape(::grpc::ServerContext* context, in GetShape() function in xla::GRPCService
H A Dgrpc_stub.cc109 Status GRPCStub::GetShape(const GetShapeRequest* request, in GetShape() function in xla::GRPCStub
/aosp_15_r20/external/armnn/include/armnn/backends/
H A DTensorHandle.hpp56 TensorShape GetShape() const override { return m_TensorInfo.GetShape(); } in GetShape() function in armnn::ConstTensorHandle
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dir_array.h237 const Shape& GetShape() const { return shape_; } in GetShape() function
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_tensor.cc66 llvm::SmallVector<tfrt::Index, 4> GetShape( in GetShape() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dshape_ops.h34 inline Status GetShape(OpKernelContext* ctx, int input_index, in GetShape() function
/aosp_15_r20/external/armnn/src/backends/aclCommon/
H A DArmComputeTensorUtils.hpp283 inline TensorShape GetShape(const arm_compute::TensorShape& shape) in GetShape() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dtest_util.cc56 std::vector<int> FlexModelTest::GetShape(int tensor_index) { in GetShape() function in tflite::flex::testing::FlexModelTest
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dconstant_map.cpp66 std::optional<c10::SymbolicShape> ConstantValueMap::GetShape( in GetShape() function in torch::jit::ConstantValueMap
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dclient.cc451 StatusOr<Shape> Client::GetShape(const GlobalData& data) { in GetShape() function in xla::Client
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Dquantization_ops_test.cc300 static StatusOr<PartialTensorShape> GetShape(const std::string& name, in GetShape() function in tensorflow::tensorrt::convert::QDQExplicitTest
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dkernel_util_test.cc84 std::vector<int> GetShape(TfLiteIntArray* dims) { in GetShape() function in tflite::__anonc351f59c0111::KernelUtilTest
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emission_utils.cc652 Shape GetShape(mlir::Value value) { in GetShape() function

12