Searched defs:buffer_shape (Results 1 – 17 of 17) sorted by relevance
34 Status ValidateUpdateShape(const TensorShape& buffer_shape, in ValidateUpdateShape()95 TensorShape buffer_shape; in Compile() local150 TensorShape buffer_shape = context->InputShape(0); in CompileTensorScatter() local
139 Status GetTensorListBufferShape(xla::XlaOp list, xla::Shape* buffer_shape) { in GetTensorListBufferShape()213 auto buffer_shape = xla::ShapeUtil::GetTupleElementShape(list_shape, 0); in GetLeadingDimForTensorList() local517 const xla::Shape& buffer_shape = in ExecuteTensorListGetItem() local
83 TensorShape buffer_shape = data_shape; in Compile() local
419 TensorShape buffer_shape; in Compile() local
73 TensorShape buffer_shape = data_shape; in Compile() local
102 const Shape& buffer_shape = in ReadDynamicShapes() local
613 const Shape& buffer_shape, in DeviceCompare()
656 const Shape& buffer_shape) { in PickBestAlgorithmNoCacheCuda()
46 auto buffer_shape = DimVector(unsqueezed.sizes()); in two_pass_reduction() local
283 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
208 const Shape& buffer_shape = in ReadDynamicShapes() local
649 const Shape& buffer_shape = instruction->operand(0)->shape(); in AddMandatoryConstraints() local
523 Status CheckBufferOffset(const Shape& buffer_shape, in CheckBufferOffset()
939 xla::Shape buffer_shape; in XLAShapeForArgument() local
1645 const Shape& buffer_shape, in CheckCompatibleShapes()