Searched defs:TensorShape (Results 1 – 4 of 4) sorted by relevance
68 const std::function<TensorShape(const TensorShape&, DataType)>* in XlaContext()
36 class TensorShape; variable
533 MAKE_SHAPE(int32, TensorShape) in MAKE_SHAPE() argument
403 class TensorShape(object): class