Searched defs:GetShape (Results 1 – 10 of 10) sorted by relevance
51 std::vector<int> GetShape(TfLiteIntArray* dims) { in GetShape() function in tflite::__anonc6585c430111::KernelUtilTest
136 const std::vector<int>& GetShape(int id) { return tensor_data_.at(id).shape; } in GetShape() function
183 const Shape& GetShape() const { in GetShape() function
362 StatusOr<Shape> Client::GetShape(const GlobalData& data) { in GetShape() function in xla::Client
224 StatusOr<std::unique_ptr<Shape>> ComputationBuilder::GetShape( in GetShape() function in xla::ComputationBuilder
307 std::unique_ptr<Shape> LocalComputationBuilder::GetShape( in GetShape() function in xla::swig::LocalComputationBuilder
639 def GetShape(self, operand): member in ComputationBuilder
1287 tensorflow::Status Service::GetShape(const GetShapeRequest* arg, in GetShape() function in xla::Service
1467 StatusOr<Shape> UserComputation::GetShape(const ComputationDataHandle& handle) { in GetShape() function in xla::UserComputation
977 TensorShapeProto GetShape(const string& input_name) const { in GetShape() function in tensorflow::grappler::__anon45c5d3ef0111::Conv2DProcessor