Searched refs:ConvertToMlirShape (Results 1 – 4 of 4) sorted by relevance
41 void ConvertToMlirShape(const TensorShape& input_shape,45 Status ConvertToMlirShape(const TensorShapeProto& input_shape,
162 void ConvertToMlirShape(const TensorShape& input_shape, in ConvertToMlirShape() function170 Status ConvertToMlirShape(const TensorShapeProto& input_shape, in ConvertToMlirShape() function192 TF_RETURN_IF_ERROR(ConvertToMlirShape(shape, &shape_dims)); in ConvertToMlirTensorType()
128 ConvertToMlirShape(input_shape, &shape); in ConvertTensor()
2136 TF_RETURN_IF_ERROR(ConvertToMlirShape(node_info.shape, &shape)); in InferLibFunctionType()2396 TF_RETURN_IF_ERROR(ConvertToMlirShape(node_info.shape, &shape)); in InferMainFunctionType()