Searched defs:buffer_shape (Results 1 – 13 of 13) sorted by relevance
34 Status ValidateUpdateShape(const TensorShape& buffer_shape, in ValidateUpdateShape()89 TensorShape buffer_shape; in Compile() local142 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
81 TensorShape buffer_shape = data_shape; in Compile() local
411 TensorShape buffer_shape; in Compile() local
71 TensorShape buffer_shape = data_shape; in Compile() local
564 const Shape& buffer_shape, in DeviceCompare()
396 const Shape& buffer_shape) { in PickBestAlgorithmNoCacheCuda()
153 llvm::SmallVector<int64_t, 8> buffer_shape; in CreateInitBufferValue() local
336 llvm::SmallVector<int64_t, 8> buffer_shape; in HandleTensorArraySplitV3Op() local
213 const Shape& buffer_shape = in ReadDynamicShapes() local
536 const Shape& buffer_shape = instruction->operand(0)->shape(); in AddMandatoryConstraints() local
1486 const xla::Shape& buffer_shape = xla::ShapeUtil::GetSubshape( in Initialize() local
913 xla::Shape buffer_shape; in XLAShapeForArgument() local