Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_expression_test.cc93 TEST_F(XlaExpressionTest, GetShape) { in TEST_F() argument
Dxla_expression.cc141 xla::StatusOr<TensorShape> XlaExpression::GetShape() const { in GetShape() function in tensorflow::XlaExpression
/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.h219 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/lite/kernels/
Dkernel_util_test.cc53 std::vector<int> GetShape(TfLiteIntArray* dims) { in GetShape() function in tflite::__anon5b8f934f0111::KernelUtilTest
Dtest_util.h238 const std::vector<int>& GetShape(int id) { return tensor_data_.at(id).shape; } in GetShape() function
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc450 StatusOr<Shape> Client::GetShape(const GlobalData& data) { in GetShape() function in xla::Client
Dxla_builder.cc104 StatusOr<Shape> XlaBuilder::GetShape(const XlaOp& op) const { in GetShape() function in xla::XlaBuilder
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc1119 Status Service::GetShape(const GetShapeRequest* arg, GetShapeResponse* result) { in GetShape() function in xla::Service
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc437 StatusOr<Shape> ComputationBuilder::GetShape(const LocalOp& operand) { in GetShape() function in xla::swig::ComputationBuilder
Dxla_client.py1154 def GetShape(self, operand): member in ComputationBuilder
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc1056 TensorShapeProto GetShape(const string& input_name) const { in GetShape() function in tensorflow::grappler::__anonbd4a17fb0111::Conv2DProcessor
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc1163 PyObject* PyTapeTensor::GetShape() const { in GetShape() function in PyTapeTensor