Searched defs:buffer_shape (Results 1 – 5 of 5) sorted by relevance
51 TensorShape* buffer_shape) { in GetTensorListBufferShape()60 TensorShape* buffer_shape) { in GetTensorListBufferShape()75 const TensorShape& buffer_shape, in InitializeTensorList()
32 Status ValidateUpdateShape(const TensorShape& buffer_shape, in ValidateUpdateShape()87 TensorShape buffer_shape; in Compile() local
77 TensorShape buffer_shape = data_shape; in Compile() local
678 xla::Shape buffer_shape; in XLAShapeForArgument() local
518 const Shape& buffer_shape = instruction->operand(0)->shape(); in AddMandatoryConstraints() local