Searched defs:buffer_shape (Results 1 – 16 of 16) sorted by relevance
34 Status ValidateUpdateShape(const TensorShape& buffer_shape, in ValidateUpdateShape()90 TensorShape buffer_shape; in Compile() local143 TensorShape buffer_shape = context->InputShape(0); in CompileTensorScatter() local
139 Status GetTensorListBufferShape(xla::XlaOp list, xla::Shape* buffer_shape) { in GetTensorListBufferShape()212 auto buffer_shape = xla::ShapeUtil::GetTupleElementShape(list_shape, 0); in GetLeadingDimForTensorList() local515 const xla::Shape& buffer_shape = in ExecuteTensorListGetItem() local
83 TensorShape buffer_shape = data_shape; in Compile() local
413 TensorShape buffer_shape; in Compile() local
73 TensorShape buffer_shape = data_shape; in Compile() local
612 const Shape& buffer_shape, in DeviceCompare()
360 const Shape& buffer_shape) { in PickBestAlgorithmNoCacheCuda()
281 const Shape& buffer_shape = in ReadDynamicShapesOnCpu() local
153 llvm::SmallVector<int64_t, 8> buffer_shape; in CreateInitBufferValue() local
335 llvm::SmallVector<int64_t, 8> buffer_shape; in HandleTensorArraySplitV3Op() local
214 const Shape& buffer_shape = in ReadDynamicShapes() local
614 const Shape& buffer_shape = instruction->operand(0)->shape(); in AddMandatoryConstraints() local
507 Status CheckBufferOffset(const Shape& buffer_shape, in CheckBufferOffset()
922 xla::Shape buffer_shape; in XLAShapeForArgument() local
1585 xla::Shape buffer_shape = xla::ShapeUtil::GetSubshape( in Initialize() local
1599 const Shape& buffer_shape, in CheckCompatibleShapes()