Searched defs:TensorShapeToXLAShape (Results 1 – 1 of 1) sorted by relevance
103 Status TensorShapeToXLAShape(DataType dtype, in TensorShapeToXLAShape() function112 xla::Shape TensorShapeToXLAShape(xla::PrimitiveType type, in TensorShapeToXLAShape() function137 Status TensorShapeToXLAShape(DataType dtype, const TensorShape& tensor_shape, in TensorShapeToXLAShape() function145 StatusOr<xla::Shape> TensorShapeToXLAShape(DataType dtype, in TensorShapeToXLAShape() function152 xla::Shape TensorShapeToXLAShape(xla::PrimitiveType type, in TensorShapeToXLAShape() function