Searched defs:extract_shape (Results 1 – 2 of 2) sorted by relevance
1317 inline TensorShape extract_shape(T *data) in extract_shape() function1322 inline TensorShape extract_shape(ITensorInfo *data) in extract_shape() function1326 inline TensorShape extract_shape(const ITensorInfo *data) in extract_shape() function1331 inline TensorShape extract_shape(const TensorShape *data) in extract_shape() function1336 inline TensorShape extract_shape(TensorShape *data) in extract_shape() function
793 def extract_shape(*args, allow_cpu_scalar_tensors: bool) -> Optional[ShapeType]: function