Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_expression_test.cc93 TEST_F(XlaExpressionTest, GetShape) { in TEST_F() argument
Dxla_expression.cc142 xla::StatusOr<TensorShape> XlaExpression::GetShape() const { in GetShape() function in tensorflow::XlaExpression
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.cc25 llvm::Optional<llvm::ArrayRef<int64_t>> 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/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.h215 const Shape& GetShape() const { return shape_; } in GetShape() function
/external/tensorflow/tensorflow/core/kernels/
Dshape_ops.h32 inline Status GetShape(OpKernelContext* ctx, int input_index, 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.cc450 StatusOr<Shape> Client::GetShape(const GlobalData& data) { in GetShape() function in xla::Client
Dxla_builder.cc153 StatusOr<Shape> XlaBuilder::GetShape(XlaOp op) const { in GetShape() function in xla::XlaBuilder
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc182 DenseElementsAttr GetShape(Value output_val) { in GetShape() function
/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util_test.cc57 std::vector<int> GetShape(TfLiteIntArray* dims) { in GetShape() function in tflite::__anondc1314510111::KernelUtilTest
Dtest_util.h283 const std::vector<int>& GetShape(int id) { return tensor_data_.at(id).shape; } in GetShape() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc1140 Status Service::GetShape(const GetShapeRequest* arg, GetShapeResponse* result) { in GetShape() function in xla::Service
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py737 def GetShape(self, operand): member in ComputationBuilder
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc1106 TensorShapeProto GetShape(const string& input_name) const { in GetShape() function in tensorflow::grappler::__anon9cd52bfd0111::Conv2DProcessor
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc1329 PyObject* PyTapeTensor::GetShape() const { in GetShape() function in PyTapeTensor