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() function145 Status TensorShapeToXLAShape(DataType dtype, const TensorShape& tensor_shape, in TensorShapeToXLAShape() function153 xla::Shape TensorShapeToXLAShape(xla::PrimitiveType type, in TensorShapeToXLAShape() function