Searched refs:GetReturnValueShape (Results 1 – 5 of 5) sorted by relevance
181 StatusOr<Shape> GetReturnValueShape() const;225 StatusOr<Shape> GetReturnValueShape();
767 result_shape = _wrap_shape(self.computation.GetReturnValueShape())797 def GetReturnValueShape(self): member in Computation798 return _wrap_shape(self._c_computation.GetReturnValueShape())1157 def GetReturnValueShape(self): member in ComputationBuilder1158 return _wrap_shape(self._client.GetReturnValueShape())
279 %unignore xla::swig::Computation::GetReturnValueShape;292 %unignore xla::swig::ComputationBuilder::GetReturnValueShape;
404 StatusOr<Shape> Computation::GetReturnValueShape() const { in GetReturnValueShape() function in xla::swig::Computation441 StatusOr<Shape> ComputationBuilder::GetReturnValueShape() { in GetReturnValueShape() function in xla::swig::ComputationBuilder
127 self.assertEqual(c.GetShape(root), c.GetReturnValueShape())133 self.assertEqual(c.GetShape(root), c.GetReturnValueShape())