Home
last modified time | relevance | path

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

12

/external/armnn/include/armnn/
DTensor.hpp187 const TensorShape& GetShape() const { return m_Shape; } in GetShape() function in armnn::TensorInfo
188 TensorShape& GetShape() { return m_Shape; } in GetShape() function in armnn::TensorInfo
284 const TensorShape& GetShape() const { return m_Info.GetShape(); } in GetShape() function in armnn::BaseTensor
285 TensorShape& GetShape() { return m_Info.GetShape(); } in GetShape() function in armnn::BaseTensor
/external/armnn/src/backends/cl/
DClTensorHandle.hpp82 TensorShape GetShape() const override in GetShape() function in armnn::ClTensorHandle
212 TensorShape GetShape() const override in GetShape() function in armnn::ClSubTensorHandle
/external/armnn/src/backends/neon/
DNeonTensorHandle.hpp94 TensorShape GetShape() const override in GetShape() function in armnn::NeonTensorHandle
303 TensorShape GetShape() const override in GetShape() function in armnn::NeonSubTensorHandle
/external/armnn/src/dynamic/sample/
DSampleTensorHandle.hpp44 armnn::TensorShape GetShape() const override in GetShape() function in sdb::SampleTensorHandle
/external/armnn/src/backends/reference/
DRefTensorHandle.hpp44 TensorShape GetShape() const override in GetShape() function in armnn::RefTensorHandle
/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/armnn/src/backends/backendsCommon/
DCpuTensorHandle.hpp53 TensorShape GetShape() const override { return m_TensorInfo.GetShape(); } in GetShape() function in armnn::ConstCpuTensorHandle
/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/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.h233 const Shape& GetShape() const { return shape_; } in GetShape() function
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_tensor.cc66 llvm::SmallVector<ssize_t, 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.hpp244 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/service/gpu/
Dir_emission_utils.cc367 Shape FusionLayoutAnalysis::GetShape(mlir::Value value) const { in GetShape() function in xla::gpu::FusionLayoutAnalysis
911 Shape GetShape(mlir::Value value) { in GetShape() function
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc450 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::__anon4b5f7ed20111::KernelUtilTest
Dtest_util.h471 const std::vector<int>& GetShape(int id) { return tensor_data_.at(id).shape; } in GetShape() function
/external/tensorflow/tensorflow/core/ir/types/
Ddialect.cc412 llvm::Optional<llvm::ArrayRef<int64_t>> GetShape(Value value) { in GetShape() function
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc146 Value GetShape(Value input, Location loc, PatternRewriter& rewriter) { in GetShape() function
Doptimize.cc309 DenseElementsAttr GetShape(Value output_val) { in GetShape() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc1090 Status Service::GetShape(const GetShapeRequest* arg, GetShapeResponse* result) { in GetShape() function in xla::Service
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc1109 TensorShapeProto GetShape(const string& input_name) const { in GetShape() function in tensorflow::grappler::__anonb6f45ffe0111::Conv2DProcessor

12