Home
last modified time | relevance | path

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

12

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

12